소스 검색

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