yjh0410 2 gadi atpakaļ
vecāks
revīzija
fda2137ee2
1 mainītis faili ar 1 papildinājumiem un 0 dzēšanām
  1. 1 0
      models/yolov1/loss.py

+ 1 - 0
models/yolov1/loss.py

@@ -32,6 +32,7 @@ class Criterion(object):
 
 
     def loss_bboxes(self, pred_box, gt_box):
     def loss_bboxes(self, pred_box, gt_box):
         # regression loss
         # regression loss
+        print(pred_box, gt_box)
         ious = get_ious(pred_box,
         ious = get_ious(pred_box,
                         gt_box,
                         gt_box,
                         box_mode="xyxy",
                         box_mode="xyxy",