Rename test.py to val.py (#4000)
This commit is contained in:
1 parent
8ee9fd1505
commit
720aaa65c8
11 files changed
+77
-78
No files matched your search
+1
-1
@@ -633,7 +633,7 @@ def apply_classifier(x, model, img, im0):
|
||||
for j, a in enumerate(d): # per item
|
||||
cutout = im0[i][int(a[1]):int(a[3]), int(a[0]):int(a[2])]
|
||||
im = cv2.resize(cutout, (224, 224)) # BGR
|
||||
# cv2.imwrite('test%i.jpg' % j, cutout)
|
||||
# cv2.imwrite('example%i.jpg' % j, cutout)
|
||||
|
||||
im = im[:, :, ::-1].transpose(2, 0, 1) # BGR to RGB, to 3x416x416
|
||||
im = np.ascontiguousarray(im, dtype=np.float32) # uint8 to float32
|
||||
|
||||
Reference in new issue
Block a user