Zzh-tju / ultralytics-YOLOv3-Cluster-NMS

Distance-IoU Loss: Faster and Better Learning for Bounding Box Regression (AAAI 2020)
GNU General Public License v3.0
43 stars 12 forks source link

Which method in utils.py/non_max_suppression() activate "Merge + DIoU + Torchvision NMS" ? #5

Open Mechazo11 opened 3 years ago

Mechazo11 commented 3 years ago

Dear @Zzh-tju,

Firstly, thank you very much for publishing the GPU accelerated NMS codes. I am planning on implementing your work on a real-time people detection system using YOLO on a jetson board.

In your readme.file, you mentioned about a NMS method called "Merge+DIoU+TorchVision". Which "method" string in the code activates this NMS mode?

With best, @Mechazo11