Pytorch实战 | 第4天：猴痘病识别

本周的代码相对于上周增加指定图片预测与保存并加载模型这个两个模块，在学习这个两知识点后，时间有余的同学请自由探索更佳的模型结构以提升模型是识别准确率，模型的搭建是深度学习程度的重点。

🏡 我的环境：

• 语言环境：Python3.8
• 编译器：Jupyter Lab
• 深度学习环境：Pytorch
• 数据集：见文末
  

  文章目录

  • 一、 前期准备
  • • 1. 设置GPU
    • 2. 导入数据
    • 3. 划分数据集
  • 二、构建简单的CNN网络
  • 三、 训练模型
  • • 1. 设置超参数
    • 2. 编写训练函数
    • 3. 编写测试函数
    • 4. 正式训练
  • 四、 结果可视化<