pytorch图像分类实战 之迁移训练

前言

第三次任务是迁移训练自己的图像分类模型，就是根据自己的数据集来训练图像分类模型。数据集使用的是样例数据集，代码也是教程的代码模板。代码教程：https://github.com/TommyZihao/Train_Custom_Dataset 。

1、环境与数据集

环境配置与上一节一样。

数据集下载：

# 下载数据集压缩包
!wget https://zihao-openmmlab.obs.cn-east-3.myhuaweicloud.com/20220716-mmclassification/dataset/fruit30/fruit30_split.zip
# 解压
!unzip fruit30_split.zip >> /dev/null

2、实验结果部分展示

总结

本次学习主要是根据教程走了一遍流程。根据视频教程了解了每一步的一些具体细节，按照环境配置、准备数据集、图像预处理、载入并处理数据集、配置训练、模型测试等步骤来完成图像的迁移训练，这个过程我了解了很多新知识，并根据视频和教程，结合在云平台的实验，我有了很多收获。