plot_images() scale bug fix (#1566)

fix bugs in plot_images
2020-12-01 18:29:59 +08:00 · 2020-12-01 18:29:59 +08:00 · 577f298d9b
parent b6ed1104a6
commit 577f298d9b
1 changed files with 6 additions and 3 deletions
--- a/utils/plots.py
+++ b/utils/plots.py
@ -141,9 +141,12 @@ 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.shape[1] and boxes.max() <= 1:  # if normalized
-                boxes[[0, 2]] *= w  # scale to pixels
-                boxes[[1, 3]] *= h
+            if boxes.shape[1]:
+                if boxes.max() <= 1:  # if normalized
+                    boxes[[0, 2]] *= w  # scale to pixels
+                    boxes[[1, 3]] *= h
+                elif scale_factor < 1: # absolute coords need scale if image scales
+                    boxes *= scale_factor
            boxes[[0, 2]] += block_x
            boxes[[1, 3]] += block_y
            for j, box in enumerate(boxes.T):