PyImageSearch / imutils

A series of convenience functions to make basic image processing operations such as translation, rotation, resizing, skeletonization, and displaying Matplotlib images easier with OpenCV and Python.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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,