上传者: 42116596
|
上传时间: 2021-12-28 16:07:35
|
文件大小: 278KB
|
文件类型: -
糖尿病视网膜病变竞赛的解决方案
这是Kaggle的竞赛,您的任务是将每个人的眼部检查分类为5种不同程度的糖尿病导致的疾病。
这是我用来处理原始图像的代码的存储库,即卷积神经网络模型(使用keras构建)。 它主要基于论坛中提供的一个基准。 执行以下步骤:
仅使用普通图像处理到256X256,未使用其他方式调整颜色等。
通过增加1、2、3和4类来平衡不同类的图片。
使用过的VGG风格架构,使用开普勒K20c GPU,以10个时期进行训练,批量大小为32。 它运行约2天。
由于输出是有序的(疾病的阶段),因此不作为分类问题运行,而是作为回归问题运行。
将原始输出转换为疾病阶段标签。 天真的,我们可以转换到接近阶段。 但是,根据原始数据集的比例对原始分数进行排名会产生更好的Kappa分数。
最后,该模型在私有数据集上得出的Kappa为0.38,由于辍学,该结果在公共得分上接近0.3