nanodet 训练细节解读

Posted 2023-01-15 蓝羽飞鸟

目录

• • 读取数据集 和 annotation
  • 训练
  • • 优化器设置
    • train_step
    • • 数据过模型，aux_head用在训练中
      • 网络输出进一步得到预测box坐标和class
      • Cost Matrix计算
      • 结合IOU 和 Cost Matrix 得到目标box坐标
    • loss计算
    • • QFL loss
      • GIOU loss
      • DFL loss
    • Optimizer

读取数据集 和 annotation

用的是COCO数据集

annotation如下，用的是instances_train2017.json