search

FeiYull / TensorRT-Alpha

🔥🔥🔥TensorRT for YOLOv8、YOLOv8-Pose、YOLOv8-Seg、YOLOv8-Cls、YOLOv7、YOLOv6、YOLOv5、YOLONAS......🚀🚀🚀CUDA IS ALL YOU NEED.🍎🍎🍎
GNU General Public License v2.0
1.26k stars 196 forks source link

对不同的类也进行NMS #52

Closed CF-chen-feng-CF closed 7 months ago

CF-chen-feng-CF commented 8 months ago

使用yolov5推理时，只进行了同一类的NMS，那如何对不同的类也进行NMS勒

FeiYull commented 8 months ago

@CF-chen-feng-CF 需要修改核函数：https://github.com/FeiYull/TensorRT-Alpha/blob/3caad3caba40517ac9df6f4fcb3c1a6dcf54477d/utils/kernel_function.cu#L539

CF-chen-feng-CF commented 8 months ago

@FeiYull 修改核函数后已解决，感谢 if (i == dx //|| pcurrent[5] != pitem[5]//) continue;