Przeglądaj źródła

train YOLOv2 on VOC

yjh0410 2 lat temu
rodzic
commit
2b3f60a57c
1 zmienionych plików z 1 dodań i 1 usunięć
  1. 1 1
      models/yolov2/loss.py

+ 1 - 1
models/yolov2/loss.py

@@ -82,7 +82,7 @@ class Criterion(object):
         loss_box = loss_box.sum() / num_fgs
         
         # obj loss
-        gt_objectness[pos_masks] *= ious.clamp(0.)
+        # gt_objectness[pos_masks] *= ious.clamp(0.)
         loss_obj = self.loss_objectness(pred_obj, gt_objectness)
         loss_obj = loss_obj.sum() / num_fgs