使用预训练的CNN提取的特征作为CNN / NN的新特征

时间:2016-12-16 17:00:42

标签: machine-learning computer-vision

我使用预训练的CNN从图片中提取特征。将这些功能用作新CNN / NN的输入是否有意义?以前做过吗?我很乐意得到答案。

2 个答案:

答案 0 :(得分:1)

这称为 finetuning 。它非常常用。通常,人们会删除VGG或类似网络的最后几层,添加适合任务的图层,并根据新数据训练网络。

请参阅:

答案 1 :(得分:0)

This link甚至可以向您展示如何在不编写任何代码行的情况下使用tensorflow进行操作。当您拥有可视化数据但样本非常少时非常有用。

我成功地使用它来比项目中的同学更好地获得结果,因为他们尝试的任何架构都没有足够的数据。如果您之前使用过张量流,则需要大约1小时才能使用,如果您没有使用过一天下午,则会遇到软件问题。