Explorar o código

train YOLOv2 on VOC

yjh0410 %!s(int64=2) %!d(string=hai) anos
pai
achega
2b3f60a57c
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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