递归神经网络的图像分类

Alb*_*han 5 python conv-neural-network keras tensorflow rnn

我正在尝试使用带有自定义数据的RNN寻找带有标签的图像分类。除了Mnist数据集,我找不到其他示例。像这样的使用CNN进行分类的存储库的任何帮助将不胜感激。关于使用RNN进行图像分类的任何帮助都将有所帮助。尝试替换以下教程的CNN网络。

Dav*_*rks 10

Aymericdamien那里有一些最好的例子,并且有一个将RNN与图像一起使用的例子。

https://github.com/aymericdamien/TensorFlow-示例

https://github.com/aymericdamien/TensorFlow-Examples/blob/master/notebooks/3_NeuralNetworks/recurrent_network.ipynb

该示例使用的是MNIST,但可以应用于任何图像。

但是,我要指出的是,您不太可能找到使用RNN对图像进行分类的许多示例,因为对于大多数图像处理任务而言,RNN都不如CNN。上面链接的示例是出于教育目的而不是实际目的。

现在,如果您尝试使用RNN是因为您要处理一系列图像,例如视频,那么在这种情况下,一种更自然的方法是将CNN(用于图像处理部分)与RNN(用于序列处理部分)。为此,您通常需要在某些分类任务(例如Imagenet)上对CNN进行预训练,然后通过CNN馈送图像,然后CNN的最后一层将作为RNN每个时间步的输入。然后,您将使用RNN上定义的损失函数来训练整个网络。