From 5b46d4971992714bdb21da8b9fa29cd4eaa847b5 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Tue, 1 Dec 2020 14:19:06 +0100
Subject: [PATCH] plot_images() scale bug fix (#1580)

From https://github.com/ultralytics/yolov5/pull/1566
---
 utils/plots.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/utils/plots.py b/utils/plots.py
index 8492b1a1..1bb61d67 100644
--- 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):