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

updates

This commit is contained in:
Glenn Jocher 2020-03-02 14:07:09 -08:00
parent 84371f6811
commit 44daace4ca
1 changed files with 7 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
test.py
View File

@ -223,8 +223,8 @@ if __name__ == '__main__':
opt.save_json = opt.save_json or any([x in opt.data for x in ['coco.data', 'coco2014.data', 'coco2017.data']]) opt.save_json = opt.save_json or any([x in opt.data for x in ['coco.data', 'coco2014.data', 'coco2017.data']])
print(opt) print(opt)
if opt.task == 'test': # task = 'test', 'study', 'benchmark' # task = 'test', 'study', 'benchmark'
# Test if opt.task == 'test': # (default) test normally
test(opt.cfg, test(opt.cfg,
opt.data, opt.data,
opt.weights, opt.weights,
@ -235,20 +235,18 @@ if __name__ == '__main__':
opt.save_json, opt.save_json,
opt.single_cls) opt.single_cls)
elif opt.task == 'benchmark': elif opt.task == 'benchmark': # mAPs at 320-608 at conf 0.5 and 0.7
# mAPs at 320-608 at conf 0.5 and 0.7
y = [] y = []
for i in [320, 416, 512, 608]: for i in [320, 416, 512, 608]: # img-size
for j in [0.5, 0.7]: for j in [0.5, 0.7]: # iou-thres
t = time.time() t = time.time()
r = test(opt.cfg, opt.data, opt.weights, opt.batch_size, i, opt.conf_thres, j, opt.save_json)[0] r = test(opt.cfg, opt.data, opt.weights, opt.batch_size, i, opt.conf_thres, j, opt.save_json)[0]
y.append(r + (time.time() - t,)) y.append(r + (time.time() - t,))
np.savetxt('benchmark.txt', y, fmt='%10.4g') # y = np.loadtxt('study.txt') np.savetxt('benchmark.txt', y, fmt='%10.4g') # y = np.loadtxt('study.txt')
elif opt.task == 'study': elif opt.task == 'study': # Parameter study
# Parameter study
y = [] y = []
x = np.arange(0.4, 0.9, 0.05) x = np.arange(0.4, 0.9, 0.05) # iou-thres
for i in x: for i in x:
t = time.time() t = time.time()
r = test(opt.cfg, opt.data, opt.weights, opt.batch_size, opt.img_size, opt.conf_thres, i, opt.save_json)[0] r = test(opt.cfg, opt.data, opt.weights, opt.batch_size, opt.img_size, opt.conf_thres, i, opt.save_json)[0]