bubbliiiing / yolov4-tiny-pytorch

这是一个YoloV4-tiny-pytorch的源码,可以用于训练自己的模型。
MIT License
787 stars 184 forks source link

loss weighted by num_pos #59

Open plutospp opened 2 years ago

plutospp commented 2 years ago

我在train.py看到这段代码

               for l in range(len(outputs)):
                    loss_item, num_pos = yolo_loss(l, outputs[l], targets)
                    loss_value_all  += loss_item
                    num_pos_all     += num_pos
                loss_value  = loss_value_all / num_pos_all

对正样本做反向加权的用意是? 这是原作使用的trick吗?

bubbliiiing commented 2 years ago

归一化