跳过正文
  1. 从模型结构到真实系统的项目沉淀。/

Classification:通用图像分类训练模板

该项目从 Kaggle 树叶图像分类任务出发,整理成可复用的图像分类训练模板。

  • 支持替换主流网络、自定义数据集、训练日志监控和推理可视化。
  • 以 176 类、18k+ 训练样本的细粒度树叶分类作为完整示例。
  • 结合 Label Smoothing、Focal Loss 以及 AdamW + CosineAnnealing 策略训练。

项目链接:GitHub

相关文章