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

updates

This commit is contained in:
Glenn Jocher 2020-01-09 14:07:55 -08:00
parent bb9c6e7a8f
commit 759d275017
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
utils/utils.py
View File

@ -840,13 +840,13 @@ def apply_classifier(x, model, img, im0):
d[:, :4] = xywh2xyxy(b).long() d[:, :4] = xywh2xyxy(b).long()
# Rescale boxes from img_size to im0 size # Rescale boxes from img_size to im0 size
scale_coords(img.shape[2:], d[:, :4], im0.shape) scale_coords(img.shape[2:], d[:, :4], im0[i].shape)
# Classes # Classes
pred_cls1 = d[:, 6].long() pred_cls1 = d[:, 5].long()
ims = [] ims = []
for j, a in enumerate(d): # per item for j, a in enumerate(d): # per item
cutout = im0[int(a[1]):int(a[3]), int(a[0]):int(a[2])] cutout = im0[i][int(a[1]):int(a[3]), int(a[0]):int(a[2])]
im = cv2.resize(cutout, (224, 224)) # BGR im = cv2.resize(cutout, (224, 224)) # BGR
# cv2.imwrite('test%i.jpg' % j, cutout) # cv2.imwrite('test%i.jpg' % j, cutout)