Spaces:

xiang-wuu
/

yolov5

Runtime error

glenn-jocher commited on Dec 9, 2020

Commit

84f9bb5

•

1 Parent(s): 86f4247

Normalized mosaic plotting bug fix (#1647)

Files changed (1) hide show

utils/plots.py CHANGED Viewed

@@ -141,7 +141,7 @@ def plot_images(images, targets, paths=None, fname='images.jpg', names=None, max
             conf = None if labels else image_targets[:, 6]  # check for confidence presence (label vs pred)
             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

             conf = None if labels else image_targets[:, 6]  # check for confidence presence (label vs pred)
             if boxes.shape[1]:
+                if boxes.max() <= 1.01:  # if normalized with tolerance 0.01
                     boxes[[0, 2]] *= w  # scale to pixels
                     boxes[[1, 3]] *= h
                 elif scale_factor < 1:  # absolute coords need scale if image scales