up你好！关于增加两个值（就和框的坐标一样）进行预测的问题

[15505529663]

对头的修改
self.yolo_head_P3 = nn.Conv2d(transition_channels * 8, len(anchors_mask[2]) * (5 + num_classes + 2), 1) self.yolo_head_P4 = nn.Conv2d(transition_channels * 16, len(anchors_mask[1]) * (5 + num_classes + 2), 1) self.yolo_head_P5 = nn.Conv2d(transition_channels * 32, len(anchors_mask[0]) * (5 + num_classes + 2), 1)
但是在yolo_training 里的YOLOLOSS我就没有思路该如何修改了，我想加一个MSE用来计算新加值的损失，然后对于后面的map_out, 以及yolo的部分就更没有思路，
所以想请问up有没有修改的思路 麻烦你了