This commit is contained in:
Glenn Jocher 2019-12-20 09:44:21 -08:00
parent 9420b4d4bc
commit 25580dfb84
1 changed files with 2 additions and 3 deletions

View File

@ -238,12 +238,11 @@ if __name__ == '__main__':
else:
# Parameter study
y = []
x = np.arange(0.3, 0.9, 0.02)
x = np.arange(0.3, 0.9, 0.1)
for v in x:
t = time.time()
r = test(opt.cfg, opt.data, opt.weights, opt.batch_size, opt.img_size, opt.conf_thres, v, opt.save_json)[0]
dt = [time.time() - t]
y.append(r + dt)
y.append(r + (time.time() - t,))
y = np.stack(y, 0)
# Plot