nanodet 训练细节解读

目录

    • 读取数据集 和 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

你可能感兴趣的:(源码解读系列,python,深度学习)