视觉计算辅助良品检测

• 参赛地址：雪浪制造AI挑战赛

---

版本

• xuelang (排名：32/2403)

• YOLO v1版

（仅适用有xml标注的数据，效果不好）

预训练模型：

最新保存的模型

验证集最好的模型

• Resnet152

（使用原图训练，未进行裁剪）

支持网络模型：VGG16、VGG19、ResNet18、ResNet152

结果：（均为未训练完成即提交）

模型（仅训练分类层）	原图resize的大小	结果(线上AUC)
resnet152	224	0.72
vgg19	420	0.753

---

提升思路：

• 模型全局训练

• 数据增强

• 数据裁剪

---

• 环境：

  python版本	pytorch版本
  3.5	0.3.0

• 说明：

  1、运行main.py请确保启动可视化工具visdom

  2、所有参数均在config.py中设置即可

---

项目结构

• 总结构

  

• 一般项目结构

  1、定义网络

  

  2、封装数据集

  

  3、工具类

  

  4、主函数

  

---

特别鸣谢：朱辉师兄