New way for register nms in onnx for tensorrt onnxruntime openvino

[triple-mu]

The last pr provides a basic solution for exporting onnx and modifying it.
This pr improves the last pr so that registered nms is completely dependent on pytorch. Simple yet effective!

🛠️ PR Summary

_{Made with ❤️ by Ultralytics Actions}

WARNING ⚠️ this PR is very large, summary may not cover all changes.

🌟 Summary

Detect and classify objects in images, videos, directories, and streams with YOLOv5.

📊 Key Changes

• Added segmentation and classification capabilities.
• Enhanced sources handling for detection (screen capture, list of images/streams).
• Improved download scripts for ease of use and added new datasets.
• Streamlined image and video inference with new options and better resizing.
• Provided various model format support for both detection and classification.

🎯 Purpose & Impact

• Provides users with flexible, high-accuracy object detection and classification across diverse media sources.
• These upgrades expand the use cases of YOLOv5 and enable more complex computer vision tasks.
• Improved download scripts and additional models formats lower barriers for adoption and deployment.

[glenn-jocher]

@triple-mu thanks for the PR! The easiest argument structure is to simply use an --nms arg that would be handled accordingly for formats that are nms capable.

[visualcortex-team]

This looks good @triple-mu and the process looks correct (agree with @glenn-jocher comment above).

We have found that the EfficientNMS plugin does not always work very predictably as FP16 support was only merged in recently and so does not work correctly in things like the NVIDIA DeepStream Docker images (on older TensorRT versions). It would be better to use the TensorRT BatchedNMS plugin which has been around longer and is more stable. Given you are already converting cx,cy,w,h to tlbr format it should be easy to update 👍

[zhiqwang]

FP16 support was only merged in recently

Hi @visualcortex-team , TensorRT supports EfficientNMS fp16 mode from 8.2.4+.

It would be better to use the TensorRT BatchedNMS plugin which has been around longer and is more stable.

But EfficientNMS plugin is much faster than BatchedNMS. BTW, TensorRT release 8.4 GA today.

[visualcortex-team]

Thanks @zhiqwang . I guess people will either:

• need to know that the minimum supported version of TensorRT to use EfficientNMS is 8.2.4+ (as it will not throw errors it will just produce no results) - this could be added as a warning when exporting?

or

• A flag is provided to choose which NMS to export with (BatchedNMS vs EfficientNMS)

[zhiqwang]

Hi @visualcortex-team

need to know that the minimum supported version of TensorRT to use EfficientNMS is 8.2.4+ (as it will not throw errors it will just produce no results) - this could be added as a warning when exporting?

Agreed!

A flag is provided to choose which NMS to export with (BatchedNMS vs EfficientNMS)

Of course there is no problem supporting BatchedNMS plugin from a technical point of view. Seems that TensorRT made this BatchedNMS plugin according to the TensorFlow's interface, it is very slow, there is no need to support it for me.

[visualcortex-team]

For future readers:

• the TensorRT release @zhiqwang spoke about updated the apt packages to version: tensorrt-dev/unknown 8.4.1.5-1+cuda11.6 amd64 which will work correctly with the EfficientNMS plugin.

• to decode the bounding boxes if used with Nvidia Deepstream you will need a custom decoder implementation in nvdsinfer_custombboxparser.cpp using these mappings (unfortunately none of the others work):

object.left=p_bboxes[4*i];
object.top=p_bboxes[4*i+1];
object.width=(p_bboxes[4*i+2] - object.left);
object.height= (p_bboxes[4*i+3] - object.top);

[triple-mu]

Why I can not pass CI test?
@glenn-jocher

[triple-mu]

@glenn-jocher
I am very happy that yolov5 will support dynamic batch at #8526.
At the same time, I also applied dynamic batch to the registered NMS!
--nms default set tensorrt nms into onnx or tf.js nms into model
--nms 0 or any int set onnx nms into onnx nms values is the same as max-wh in nms
--dynamic default dynamic all axes into onnx or tf.js model not support onnx for tensorrt
--dynamic 0 dynamic batch into onnx also support onnx for tensorrt

[glenn-jocher]

@triple-mu thanks for the input! We'll make sure to consider the dynamic batching and NMS options in our future development. Your feedback is greatly appreciated!