Compatibility of vectorized non_max suppression code with Ultralytics Yolov3 non max suppression code?
Mechazo11 opened this issue · comments
Dear @jrosebr1,
Thank you for making these awesome object detection codes available publicly online.
I am working on an people detection project and am using Ultralytics YoloV3 as the base code. In his general.py file, the author uses a form of NMS which takes a Pytorch tensor containing all the bounding box proposals as inputs. The code can be found here https://github.com/ultralytics/yolov3/blob/master/utils/general.py
I was wondering is your implementation of nms compatible with Ultralytic's NMS code?
With best regards,