Cassava Leaf Disease Classification:我的第一个kaggle比赛 源码

木薯叶病分类 目录 我的木薯叶病比赛学习历程。我花了整整3周的时间参加这项比赛。 目标: 对木薯叶上呈现的疾病类型进行分类。有五个不同的标签:木薯细菌枯萎病(CBB),木薯褐斑病(CBSD),木薯绿斑驳病(CGM),木薯花叶病(CMD)和健康。 挑战: 这项比赛有一些挑战。 每个班级之间的分配不平衡。标签3 CMD与其他类别之间存在巨大差异。这可能会对预测产生偏差。因此,必须采用加权损失函数或过采样。 嘈杂的标签。一幅图像中有很多错误标记的图像和多种疾病,可能会影响模型预测。为了解决这个问题,可以实现多种技术,例如标签平滑,混合,剪切混合增强。 我在截止日期前参加了比赛,因此就如何处理嘈杂的标签以及哪种模型在本次比赛中效果最好,进行了很多讨论。大多数竞争对手都采用了Efficientnet和视觉变压器(ViT),因此在尝试其他模型(例如ViT,DeiT,Hybrid Resnet和ViT)