From 3e85a4191a9f7b63081a74761d0b6987d284aea9 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 9 Apr 2019 13:19:17 +0200 Subject: [PATCH] updates --- utils/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/utils.py b/utils/utils.py index a6c15c6d..e13d095b 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -496,7 +496,7 @@ def plot_images(imgs, targets, fname='images.jpg'): for i in range(bs): boxes = xywh2xyxy(targets[targets[:, 0] == i, 2:6]).numpy().T * img_size - plt.subplot(sp, sp, i + 1).imshow(imgs[i].numpy().transpose(1, 2, 0)) + plt.subplot(sp, sp, i + 1).imshow(imgs[i].cpu().numpy().transpose(1, 2, 0)) plt.plot(boxes[[0, 2, 2, 0, 0]], boxes[[1, 1, 3, 3, 1]], '.-') plt.axis('off') fig.tight_layout()