
【MRI脑肿瘤】基于PyTorch的医学图像分类,采用三种深度学习网络及代码与数据集(含中文注释)
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本项目利用PyTorch框架进行MRI脑肿瘤图像分类,结合三种深度学习模型,并提供详细中文注释的数据集和代码。
本项目使用了一个数据集,包含155张脑部肿瘤切片MRI图像与98张正常MRI图像,并以9:1的比例划分训练集和验证集。通过三种不同的深度学习神经网络(AlexNet、LeNet 和 ResNet)实现对这些肿瘤图像的二分类任务(有无肿瘤)。其中,ResNet部分采用了迁移学习方法使用已有模型,显著提高了分类效果。
代码简洁易懂,并且在上传时已测试确保全部能够正常运行。可以在此基础上进行改进并迁移到自己的数据中;注释非常详细,在训练文件中有如下说明:
```python
该文件用于训练模型(AlexNet/LeNet)。
通过修改以下几处可切换两个网络的训练:
1、选择网络,可选LeNet或者AlexNet
net = AlexNet(num_classes=2, init_weights=True)
# net = LeNet(num_classes=2)
2、设置模型训练完成后的路径与名称
save_path = ./AlexNet.pth
# save_path = ./LeNe
```
本代码来源于本人学习过程中的测试总结,适合初学者入门。分享给大家,共同学习!
全部评论 (0)
还没有任何评论哟~


