浏览代码

train YOLOv2 on VOC

yjh0410 2 年之前
父节点
当前提交
2b3f60a57c
共有 1 个文件被更改,包括 1 次插入1 次删除
  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