-
🍨 本文为🔗365天深度学习训练营中的学习记录博客
-
🍖 原作者:K同学啊
前言
-
图像分类是深度学习入门的重要任务,而卷积神经网络(CNN)在其中发挥着关键作用。
-
本项目聚焦于猴痘病图像识别 ,使用的是包含多种皮肤病变类型的彩色图像数据集,相比 MNIST 等简单数据集更具复杂性和现实意义。
-
我们基于 PyTorch 构建了一个 CNN 模型,完成了从数据预处理、模型训练到测试预测的完整流程。
-
该项目不仅加深了我们对图像分类技术的理解,也为今后探索医学图像分析打下了基础。
代码实现
1. 数据可视化
import matplotlib.pyplot as plt
import os
from PIL import Image
# 图像文件夹路径
image_folder = r'D:\DL_Bootcamp\P4\data\Monkeypox'
# 获取所有图像文件名
image_files =