文件名称:Diabetic_Reinopathy_Detection:使用Pytorch检测糖尿病性视网膜病变
文件大小:5KB
文件格式:ZIP
更新时间:2024-05-29 03:31:15
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)