debug loss of YOLOv2

models/yolov2/loss.py

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

test.py

@@ -137,7 +137,6 @@ def test(args,
         # inference
         bboxes, scores, labels = model(x)
         print("detection time used ", time.time() - t0, "s")
-        print(bboxes)
         
         # rescale bboxes
         origin_img_size = [orig_h, orig_w]