如何使用模型

来源：8-20 训练流程代码

紫梦沁香

2021-08-10

训练好的模型该如何使用呢？
比如这个图像生成文本的模型，我该如何输入图像来生成文本呢？
虽然结果肯定不如人意，就是想知道方法
图片描述

写回答

1回答

正十七

2021-08-23

我们的课程中的模型是LSTM，其实操作比较简单，大致的code也在我们的代码库里：https://git.imooc.com/coding-259/coding-259/src/master/py3/08-image-caption/image_caption_eval.ipynb

简要来说，当遇到一张图片，首先做裁剪，让图片大小和训练数据里的设置一样，然后把图片输入给卷积神经网络，得到图片的embedding，然后再经过一个全连接层，得到一个和word embedding大小一样的embedding。

在然后就是循环操作，把图片embedding当作LSTM的第一个输入，然后得到下一个词的预测的概率分布，选择概率最高的词语，找到它的embedding，再输入给Lstm，得到下一个词的概率分布，再选择概率最大的，输入给LSTM，依此类推，就得到了生成的句子了。

深度学习之神经网络(CNN/RNN/GAN)算法原理+实战

深度学习算法工程师必学，深入理解深度学习核心算法CNN RNN GAN

2617 学习 · 935 问题

相似问题

回答 1

回答 1

回答 1

回答 1

回答 1