Remove redundant speed/study `half` argument (#3557)
Browse files
test.py
CHANGED
@@ -336,8 +336,7 @@ if __name__ == '__main__':
|
|
336 |
|
337 |
elif opt.task == 'speed': # speed benchmarks
|
338 |
for w in opt.weights if isinstance(opt.weights, list) else [opt.weights]:
|
339 |
-
test(opt.data, w, opt.batch_size, opt.img_size, 0.25, 0.45, save_json=False, plots=False,
|
340 |
-
opt=opt)
|
341 |
|
342 |
elif opt.task == 'study': # run over a range of settings and save/plot
|
343 |
# python test.py --task study --data coco.yaml --iou 0.7 --weights yolov5s.pt yolov5m.pt yolov5l.pt yolov5x.pt
|
@@ -348,7 +347,7 @@ if __name__ == '__main__':
|
|
348 |
for i in x: # img-size
|
349 |
print(f'\nRunning {f} point {i}...')
|
350 |
r, _, t = test(opt.data, w, opt.batch_size, i, opt.conf_thres, opt.iou_thres, opt.save_json,
|
351 |
-
plots=False,
|
352 |
y.append(r + t) # results and times
|
353 |
np.savetxt(f, y, fmt='%10.4g') # save
|
354 |
os.system('zip -r study.zip study_*.txt')
|
|
|
336 |
|
337 |
elif opt.task == 'speed': # speed benchmarks
|
338 |
for w in opt.weights if isinstance(opt.weights, list) else [opt.weights]:
|
339 |
+
test(opt.data, w, opt.batch_size, opt.img_size, 0.25, 0.45, save_json=False, plots=False, opt=opt)
|
|
|
340 |
|
341 |
elif opt.task == 'study': # run over a range of settings and save/plot
|
342 |
# python test.py --task study --data coco.yaml --iou 0.7 --weights yolov5s.pt yolov5m.pt yolov5l.pt yolov5x.pt
|
|
|
347 |
for i in x: # img-size
|
348 |
print(f'\nRunning {f} point {i}...')
|
349 |
r, _, t = test(opt.data, w, opt.batch_size, i, opt.conf_thres, opt.iou_thres, opt.save_json,
|
350 |
+
plots=False, opt=opt)
|
351 |
y.append(r + t) # results and times
|
352 |
np.savetxt(f, y, fmt='%10.4g') # save
|
353 |
os.system('zip -r study.zip study_*.txt')
|