search

yijingru / BBAVectors-Oriented-Object-Detection

[WACV2021] Oriented Object Detection in Aerial Images with Box Boundary-Aware Vectors
MIT License
469 stars 88 forks source link

NMS #95

Closed candyotter closed 3 years ago

candyotter commented 3 years ago

Hello, Does this use NMS and/or is there a way I can change the NMS threshold, I'm getting quite a few detections on the same object (custom dataset) and think this may be a way to reduce it.

Thanks for you help!

yijingru commented 3 years ago

You may want to check here https://github.com/yijingru/BBAVectors-Oriented-Object-Detection/blob/7efd410f5f4ded94aca986aeb0bd1292235d1314/func_utils.py#L89