zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Normalized mosaic plotting bug fix (#1647)

This commit is contained in:
Glenn Jocher 2020-12-08 18:44:13 -08:00 committed by GitHub
parent 86f4247515
commit 84f9bb5d92
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/plots.py
View File

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