DataXujing/YOLOv9

关于EfficientNMS_TRT Plugin节点插入的疑问

Closed this issue · 2 comments

您好,非常感谢您这个yolov9部署的开源库,我在阅读您的onnx_add_nms_op.py代码时遇到了一个问题,想向您请教一下。
就是在这个def create_and_add_plugin_node()函数中的这段代码:
mns_node = gs.Node(
op="EfficientNMS_TRT",
attrs=create_attrs(max_output_boxes),
inputs=[boxes_tensor, confs_tensor],
outputs=new_outputs)
我理解您这里是想创建一个EfficientNMS_TRT操作的onnx 节点,但是我在找了项目里面的.cpp或者yolo.py common.py等文件并没有找到EfficientNMS_TRT相关的计算代码,想问一下这里创建了EfficientNMS_TRT这个节点是onnx自动搜索库文件里面相关EfficientNMS_TRT的操作吗?如果是自定义的节点为啥没有计算逻辑呢?
我是c++ tensorRT部署方面的新手,如有冒犯请多多包涵。

这个节点在TensorRT中,是TensorRT官方提供的Plugin,已经注册在TensorRT pligin lib中: https://github.com/NVIDIA/TensorRT/tree/release/10.0/plugin/efficientNMSPlugin

好的,我明白了,谢谢您的回答