From 43e3bccc7366a3c9b1f929a71a53263b28d62c1c Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 20 Dec 2019 09:10:35 -0800 Subject: [PATCH] updates --- test.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/test.py b/test.py index 9eb0cf89..eed75e1d 100644 --- a/test.py +++ b/test.py @@ -237,7 +237,7 @@ if __name__ == '__main__': else: # Parameter study y = [] - x = np.arange(0.4, 0.81, 0.1) + x = np.arange(0.3, 0.9, 0.02) for v in x: y.append(test(opt.cfg, opt.data, opt.weights, opt.batch_size, opt.img_size, 0.1, v, True)[0]) y = np.stack(y, 0) @@ -247,5 +247,7 @@ if __name__ == '__main__': ax.plot(x, y[:, 2], marker='.', label='mAP@0.5') ax.plot(x, y[:, 3], marker='.', label='mAP@0.5:0.95') ax.legend() + ax.set_xlabel('nms_thr') + ax.set_ylabel('mAP') fig.tight_layout() plt.savefig('parameters.jpg', dpi=200)