这段简介可以这样写:“Kaggle-Cassava-Leaf-Disease-Classification”项目是一个针对木薯叶病进行分类的比赛代码,旨在通过图像识别技术帮助农民精准诊断作物疾病。
在Kaggle的木薯叶病分类竞赛中,“木薯叶病分类”项目取得了第256名的成绩(总排名为3900中的前7%),并获得了铜牌,比赛于二月份结束。最近整理代码后,我决定将其发布到GitHub上。(尽管整理代码是一项繁琐的工作)。在私有数据集中,我的模型得分为0.8987 。然而,在GitHub上发布的版本得分更高为0.9010 ,如果提交该解决方案,则可以进入银牌区域。
该项目的仓库包括训练和测试部分的完整代码。我使用了一些特定的技术来优化性能:
- AMP(自动混合精度)以加快模型训练速度,考虑到Kaggle平台上的GPU时间限制以及我个人使用的较慢GPU。
- 数据增强技术Data8月被证明可以提升模型的表现;我没有采用cutmix或snapmix等方法,因为这些技巧虽然耗时较长但并未显著改善性能。
- K折交叉验证($ k = 5 $)用于模型训练和评估的稳定性
- 模型集成:EfficientB4是我自己训练的一个版本,而Resnext则是从讨论区获得公开访问权限后使用的。
以上就是我对该项目的一些说明。