diff --git a/test.py b/test.py index e30f93ee..a9d69dce 100644 --- a/test.py +++ b/test.py @@ -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