目录 读取数据集 和 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 #coco_api会包含下面anns等信息 self.coco_api = COCO(ann_path) #instances_train2017.json #这个class是anaconda自带的 class COCO: def createIndex(self)