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: else:
# Parameter study # Parameter study
y = [] y = []
x = np.arange(0.3, 0.9, 0.02) x = np.arange(0.3, 0.9, 0.1)
for v in x: for v 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, v, opt.save_json)[0] 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 + (time.time() - t,))
y.append(r + dt)
y = np.stack(y, 0) y = np.stack(y, 0)
# Plot # Plot