【文件属性】:
文件名称:Diabetic_Reinopathy_Detection:使用Pytorch检测糖尿病性视网膜病变
文件大小:5KB
文件格式:ZIP
更新时间:2021-05-09 09:44:35
Python
糖尿病_视网膜病变_检测
使用Pytorch检测糖尿病性视网膜病变
关于数据集
数据集包含左眼和右眼的图像。 可以在这里找到更多信息: :
数据预处理
由于图像有噪点,因此我通过裁剪图像将其删除,同时也存在类不平衡问题,因此我通过数据扩充将其删除。
关于实施
在这里,我在Kaggle数据集上实现了糖尿病性视网膜病变检测,该存储库中有两种实现方式:
1.二进制分类:
在bin_retinet.py中,该模型可以预测一个人是否患有糖尿病性视网膜病。
2.多类分类:
模型在multi_retinet.py中预测一个人是否患有:0-无DR 1-轻度2-中等3-严重4-增殖性DR
【文件预览】:
Diabetic_Reinopathy_Detection-master
----train_multi.py(3KB)
----README.md(844B)
----train_bin.py(2KB)
----data_multi.py(965B)
----data_bin.py(890B)
----bin_retinet.py(2KB)