|
|
|
@ -5,7 +5,7 @@ from models.experimental import *
|
|
|
|
|
from utils.datasets import *
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test(data,
|
|
|
|
|
def evaluate(data,
|
|
|
|
|
weights=None,
|
|
|
|
|
batch_size=16,
|
|
|
|
|
imgsz=640,
|
|
|
|
@ -254,7 +254,7 @@ if __name__ == '__main__':
|
|
|
|
|
print(opt)
|
|
|
|
|
|
|
|
|
|
if opt.task in ['val', 'test']: # run normally
|
|
|
|
|
test(opt.data,
|
|
|
|
|
evaluate(opt.data,
|
|
|
|
|
opt.weights,
|
|
|
|
|
opt.batch_size,
|
|
|
|
|
opt.img_size,
|
|
|
|
@ -272,7 +272,7 @@ if __name__ == '__main__':
|
|
|
|
|
y = [] # y axis
|
|
|
|
|
for i in x: # img-size
|
|
|
|
|
print('\nRunning %s point %s...' % (f, i))
|
|
|
|
|
r, _, t = test(opt.data, weights, opt.batch_size, i, opt.conf_thres, opt.iou_thres, opt.save_json)
|
|
|
|
|
r, _, t = evaluate(opt.data, weights, opt.batch_size, i, opt.conf_thres, opt.iou_thres, opt.save_json)
|
|
|
|
|
y.append(r + t) # results and times
|
|
|
|
|
np.savetxt(f, y, fmt='%10.4g') # save
|
|
|
|
|
os.system('zip -r study.zip study_*.txt')
|
|
|
|
|