返回列表

6th place solution , yolov5m6/s6, trust cv, post classification.

476. TensorFlow - Help Protect the Great Barrier Reef | tensorflow-great-barrier-reef

开始: 2021-11-22 结束: 2022-02-14 环境监测 数据算法赛
第6名解决方案:YOLOv5m6/s6,相信CV,后处理分类

第6名解决方案:YOLOv5m6/s6,相信CV,后处理分类

作者: KKY | 排名: 第6名 | 发布时间: 2022-02-15

恭喜所有相信自己的 CV(交叉验证)分数的团队,Private Board(私有排行榜)的排名发生了巨大的震荡。感谢我的队友 @khyeh0719@trushk,我在这次比赛中经历了一段很棒的旅程。

流程架构

pipeline

相信你的 CV(交叉验证)

我们按 video_id 划分数据,并在整个 OOF(Out-of-Fold)数据上计算 F2 分数,发现了一些有趣的事情:

  1. CV 显示,当推理图像分辨率 = 训练图像分辨率时,模型效果很好;但 LB(Public Leaderboard)需要 X2~X2.5 的推理分辨率才能获得不错的分数。
  2. 跟踪技术(Track)在 CV 和 LB 上都提高了约 0.002 的分数。
  3. 后处理分类在 CV 和 LB 上都提高了约 0.006~0.01 的分数。
  4. 使用低置信度进行推理并结合 WBF(加权框融合)集成,给我们带来了约 0.02 的提升。

最终我们基于以下策略选择了 4 个提交结果:

  1. 最佳 CV 提交(正常分辨率) -> LB 0.616 / Private 0.723
  2. 最佳 LB 提交 -> LB 0.716 / Private 0.692 (巨大震荡)
  3. 最佳 CV X1.5 分辨率 -> LB 0.668 / Private 0.727
  4. 最佳 CV X2.0 分辨率 -> LB 0.700 / Private 0.711

最佳 CV 模型

  1. 后处理分类使用 swin-transformer
  2. 跟踪使用 nofair 公共设置
  3. WBF 集成
  4. 单模型如下表所示:
名称 训练配置 推理配置
s1 yolov5m6_e5_bs1_lr01_img3000_public sz3600_nmsconf0.01_nmsiou0.2
s3 yolov5m6_e5_bs1_lr01_img3000_public sz2412_nmsconf0.01_nmsiou0.2
s15 yolov5s6_e5_bs2_lr01_img3000_public sz3000_nmsconf0.01_nmsiou0.2
s61 yolov5m6_e10_bs8_lr01_img3600_cp_nb8 sz3600_conf0.01_iou0.2
s62 yolov5s6_e10_bs8_lr01_img3000_cp_nb8 sz3000_conf0.01_iou0.2
s63 yolov5m6_e10_bs8_lr01_img3600_cp_nb8 sz2412_conf0.01_iou0.2
同比赛其他方案