search

xingyizhou / CenterNet2

Two-stage CenterNet
Apache License 2.0
1.2k stars 188 forks source link

CenterNet topk points #32

Open sh7jacobi opened 3 years ago

sh7jacobi commented 3 years ago

📚 Documentation

Hello, I would like to set topk quantity for Center points on validation. How should I do this?

xingyizhou commented 3 years ago

Hi, You can modify MODEL.CENTERNET.POST_NMS_TOPK_TEST in the config.