glenn-jocher
commited on
Commit
•
4f9718a
1
Parent(s):
a0e1504
Pass `--device` for `--task study` (#5118)
Browse files
val.py
CHANGED
@@ -346,7 +346,7 @@ def main(opt):
|
|
346 |
for i in x: # img-size
|
347 |
print(f'\nRunning {f} point {i}...')
|
348 |
r, _, t = run(opt.data, weights=w, batch_size=opt.batch_size, imgsz=i, conf_thres=opt.conf_thres,
|
349 |
-
iou_thres=opt.iou_thres, save_json=opt.save_json, plots=False)
|
350 |
y.append(r + t) # results and times
|
351 |
np.savetxt(f, y, fmt='%10.4g') # save
|
352 |
os.system('zip -r study.zip study_*.txt')
|
|
|
346 |
for i in x: # img-size
|
347 |
print(f'\nRunning {f} point {i}...')
|
348 |
r, _, t = run(opt.data, weights=w, batch_size=opt.batch_size, imgsz=i, conf_thres=opt.conf_thres,
|
349 |
+
iou_thres=opt.iou_thres, device=opt.device, save_json=opt.save_json, plots=False)
|
350 |
y.append(r + t) # results and times
|
351 |
np.savetxt(f, y, fmt='%10.4g') # save
|
352 |
os.system('zip -r study.zip study_*.txt')
|