Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-04-09 13:19:17 +02:00
parent 2e74f4be41
commit 3e85a4191a
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/utils.py
View File

@ -496,7 +496,7 @@ def plot_images(imgs, targets, fname='images.jpg'):
for i in range(bs): for i in range(bs):
boxes = xywh2xyxy(targets[targets[:, 0] == i, 2:6]).numpy().T * img_size 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.plot(boxes[[0, 2, 2, 0, 0]], boxes[[1, 1, 3, 3, 1]], '.-')
plt.axis('off') plt.axis('off')
fig.tight_layout() fig.tight_layout()