迁移学习是如何解决图像分类问题的

更新时间:02-08 教程 由 凉人心 分享

1卷积网络当做特征提取器。使用在ImageNet上预训练的网络,去掉最后的全连接层,剩余部分当做特征提取器(例如AlexNet在最后分类器前,是4096维的特征向量)。这样提取的特征叫做CNNcodes。得到这样的特征后,可以使用线性分类器(LinerSVM、Softmax等)来分类图像。

2Fine-tuning卷积网络。替换掉网络的输入层(数据),使用新的数据继续训练。Fine-tune时可以选择fine-tune全部层或部分层。通常,前面的层提取的是图像的通用特征(genericfeatures)(例如边缘检测,色彩检测),这些特征对许多任务都有用。后面的层提取的是与特定类别有关的特征,因此fine-tune时常常只需要Fine-tuning后面的层。

关注中公优就业,学习更多深度学习知识。

声明:关于《迁移学习是如何解决图像分类问题的》以上内容仅供参考,若您的权利被侵害,请联系13825271@qq.com
本文网址:http://www.25820.com/tutorial/14_2162622.html