零基础实战迁移学习VGG16解决图像分类问题

1 前言

本文涉及到的代码均已开源,读者可自行下载学习,实践大于理论!
什么是迁移学习?简单的理解就是使用一些已经训练好的模型迁移到类似的新的问题进行使用,而不必对新问题重新建模,从头训练和优化参数。这些训练好的模型同时包含了优化好的参数,在使用的时候只需要做一些简单的调整就可以应用到新问题中了。 本文需要解决的问题使用了迁移过来的VGG16模型,本文最终会得到一个能对猫狗图片进行辨识的CNN(卷积神经网络),测试集用来验证我的模型是否能够很好的工作。
代码已经上传到Github,读者可自行下载训练和预测,本文环境基于Colab pro,记得Star~
下载链接:https://github.com/lixiang007666/TransferLearning-pytorch
数据集链接:

猜你喜欢

转载自blog.csdn.net/weixin_43838785/article/details/120572087