视网膜光学相干断层扫描分类
Kaggle视网膜OCT图像的Pytorch实现
模型
结果
先决条件
火炬
入门
安装
从安装PyTorch和依赖项
从源头安装Torch视觉。
git clone https://github.com/pytorch/vision
cd vision
python setup.py install
视网膜OCT图像数据集
要训练模型,请修改train.py中的数据集路径
data_dir = ' dataset path '
火车
训练模型:
python train.py
测试
测试模型:
python test.py
2023-04-01 22:06:12
2MB
Python
1