上传者: 42140625
|
上传时间: 2021-11-11 15:22:34
|
文件大小: 5KB
|
文件类型: -
糖尿病_视网膜病变_检测
使用Pytorch检测糖尿病性视网膜病变
关于数据集
数据集包含左眼和右眼的图像。 可以在这里找到更多信息: :
数据预处理
由于图像有噪点,因此我通过裁剪图像将其删除,同时也存在类不平衡问题,因此我通过数据扩充将其删除。
关于实施
在这里,我在Kaggle数据集上实现了糖尿病性视网膜病变检测,该存储库中有两种实现方式:
1.二进制分类:
在bin_retinet.py中,该模型可以预测一个人是否患有糖尿病性视网膜病。
2.多类分类:
模型在multi_retinet.py中预测一个人是否患有:0-无DR 1-轻度2-中等3-严重4-增殖性DR