本节使用预训练好的ResNet-34模型来完成Kaggle竞赛中的犬种识别挑战,这里会用到模型微调的技巧,复用预训练的模型是大型比赛常常会用到的方法。

实验内容

  1. 整理数据集
  2. 图像增强
  3. 读取数据
  4. 定义模型
  5. 定义训练函数
  6. 调参
  7. 在完整数据及上训练模型
  8. 对测试集分类并提交结果
Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐