深度学习pytorch实战五:基于ResNet34迁移学习的方法图像分类篇自建花数据集图像分类(5类)超详细代码
1.数据集简介2.模型相关知识3.split_data.py——训练集与测试集划分4.model.py——定义ResNet34网络模型5.train.py——加载数据集并训练,训练集计算损失值loss,测试集计算accuracy,保存训练好的网络参数6.predict.py——利用训练好的网络参数后,用自己找的图像进行分类测试一、数据集简介1.自建数据文件夹首先确定这次分类种类,采用爬虫、官网数据