kaggle糖尿病性视网膜病
一般的
该存储库包含AI眼睛团队在Kaggle的糖尿病性视网膜病变检测比赛中的工作。 此README.md文件包含有关如何正确运行算法的一些信息。
单一模型的管道
python train.py (使用data / processed文件夹中存在的train / test分割!这很重要!)
这将训练模型并将模型,kappa图和最佳权重放置在models/文件夹中。
python optimize_threshold.py
这需要模型,最佳权重,验证拆分和真实标签。
它将用于验证集的最佳阈值输出到models//optimal_thresholds 。
python predict.py
这需要测试集图像,样品提交,模型,最佳权重和最佳阈值。
它将测试集的预测输出到
2021-09-21 21:18:31
223KB
Python
1