Mosaic plots bug fix (ultralytics#1526)

stijns · Nov 26, 2020 · 2c3efa4 · 2c3efa4
1 parent 12499f1
commit 2c3efa4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/utils/plots.py b/utils/plots.py
@@ -140,6 +140,9 @@ def plot_images(images, targets, paths=None, fname='images.jpg', names=None, max
             labels = image_targets.shape[1] == 6  # labels if no conf column
             conf = None if labels else image_targets[:, 6]  # check for confidence presence (label vs pred)
 
+            if boxes.max() <= 1:  # if normalized
+                boxes[[0, 2]] *= w  # scale to pixels
+                boxes[[1, 3]] *= h
             boxes[[0, 2]] += block_x
             boxes[[1, 3]] += block_y
             for j, box in enumerate(boxes.T):