This commit is contained in:
Glenn Jocher 2019-07-09 18:16:35 +02:00
parent 5b0ba6d7b2
commit 7b8a134a0b
1 changed files with 1 additions and 1 deletions

View File

@ -627,7 +627,7 @@ def plot_images(imgs, targets, paths=None, fname='images.jpg'):
boxes[[0, 2]] *= w boxes[[0, 2]] *= w
boxes[[1, 3]] *= h boxes[[1, 3]] *= h
plt.subplot(ns, ns, i + 1).imshow(imgs[i].transpose(1, 2, 0)) plt.subplot(ns, ns, i + 1).imshow(imgs[i].transpose(1, 2, 0))
plt.plot(boxes[[0, 2, 2, 0, 0]], boxes[[1, 1, 3, 3, 1]], 'w.-') plt.plot(boxes[[0, 2, 2, 0, 0]], boxes[[1, 1, 3, 3, 1]], '.-')
plt.axis('off') plt.axis('off')
if paths is not None: if paths is not None:
s = Path(paths[i]).name s = Path(paths[i]).name