update based on Tianxiaomo#553

Blyncs · Feb 13, 2023 · 9fbc868 · 9fbc868
1 parent e709522
commit 9fbc868
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -208,7 +208,7 @@ def build_target(self, pred, labels, batchsize, fsize, n_ch, output_id):
             truth_box[:n, 0] = truth_x_all[b, :n]
             truth_box[:n, 1] = truth_y_all[b, :n]
 
-            pred_ious = bboxes_iou(pred[b].view(-1, 4), truth_box, xyxy=False)
+            pred_ious = bboxes_iou(pred[b].reshape(-1, 4), truth_box, xyxy=False)
             pred_best_iou, _ = pred_ious.max(dim=1)
             pred_best_iou = (pred_best_iou > self.ignore_thre)
             pred_best_iou = pred_best_iou.view(pred[b].shape[:3])