使用神经网络为图像生成标题( 二 )

正如您在下面看到的,如果执行上面的代码,您将看到我们的图像特性只是一个形状-(18432,)的numpy数组 。
image_feature_dictionary[list(image_feature_dictionary. Keys())[0]].shape (18432,)接下来,我们将开发用于为图像生成标题的LSTM网络(RNN) 。
用于生成标题的LSTM文本生成是LSTM网络中最流行的应用之一 。LSTM单元格(LSTM网络的基本构建块)能够根据前一层的输出生成输出,即它保留前一层(内存)的输出,并使用该内存生成(预测)序列中的下一个输出 。
对于我们的数据集,我们为每张图片设置了5个标题,即总共40k个标题 。
让我们看看我们的数据集-

使用神经网络为图像生成标题

文章插图
 
1. 


推荐阅读