Mosaic plots bug fix (#1526)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2020-11-26 14:02:22 +01:00
1 parent 12499f1c01
commit 2c3efa430b
1 file changed
+3
+3
View File
@@ -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):