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

updates

This commit is contained in:
Glenn Jocher 2019-12-20 09:10:35 -08:00
parent 442dbb6acf
commit 43e3bccc73
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test.py
View File

@ -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)