2016 - 2024

感恩一路有你

tensorflow框架 迁移学习:如何使用TensorFlow机器学习对图像进行分类?

浏览量:3125 时间:2021-03-13 09:37:13 作者:admin

迁移学习:如何使用TensorFlow机器学习对图像进行分类?

总之,它可以分为两种方式:一种是用预训练模型作为特征抽取器,然后对模型进行训练和调整;另一种是通过精细调整达到训练分类器的目的,预训练模型与图像增强相结合。

以vgg16模型为例,我们下载了vgg16的预训练模型,可以看到vgg16模型分类器中与分类器相关的最后一个分类部分已经被删除。利用vgg-16模型作为特征提取工具,对每一层图像进行冻结,提取出待分类的瓶颈层特征,即vgg=get瓶颈特征(vgguModel,trainuIMGSuScale)

validationuFeaturesvgg=getu瓶颈特征(vgguModel,validationimgsuScale)

然后建立一个简单的模型分类器和提取的训练特征作为模型训练的输入参数,通过迭代训练得到一个简单的图像分类器。

第二种方法是以vgg-16为例。首先对前三层进行冻结,将4-5层设置为可训练层,然后结合图像分割,通过连续训练和迭代优化后两层的参数,得到更好的图像分类器。

综合评价后,第二种方法通常比第一种方法更有效。

tensorflow框架 tensorflow简单图片分类 tensorflow到底是什么

版权声明:本文内容由互联网用户自发贡献,本站不承担相关法律责任.如有侵权/违法内容,本站将立刻删除。