apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Normalized mosaic plotting bug fix (#1600)

Browse Source
This commit is contained in:
Glenn Jocher 2020-12-08 18:46:33 -08:00 committed by GitHub
parent 8d236eea3c
commit 6b1fe3e9dd
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