【PyTorch深度学习项目实战100例】—— 基于ShuffleNet实现中草药识别任务 | 第32例

前言

大家好，我是阿光。

本专栏整理了《PyTorch深度学习项目实战100例》，内包含了各种不同的深度学习项目，包含项目原理以及源码，每一个项目实例都附带有完整的代码+数据集。

正在更新中~ ✨

🚨 我的项目环境：

• 平台：Windows10
• 语言环境：python3.7
• 编译器：PyCharm
• PyTorch版本：1.8.1

💥 项目专栏：【PyTorch深度学习项目实战100例】

---

一、基于ResNet101实现猴痘病毒识别任务

中医药是“中国古代科学的瑰宝”，是中华民族的伟大创造。在推进健康中国的进程中，国务院明确了加快推进中医药现代化、产业化的方向，所以本项目在研究猴痘病毒识别分类过程中，借助人工智能算法的优势，开展基于深度神经网络的图像分类实验。借助ShuffleNet模型进行迁移学习，经迭代10次后分类准确率达到95.23%。

二、数据集介绍

本案例使用的数据集是一个中草药分类的数据集，包含枸杞、百合、党参、槐花、金银花五种类别。

数据集划分为训练集：