PPYOLOE_Tensorrt icon indicating copy to clipboard operation
PPYOLOE_Tensorrt copied to clipboard

EfficientNMS问题

Open freecarving opened this issue 3 years ago • 4 comments

请教一下作者,EfficientNMS_trt代码中NMS是加在 if(WEIGHTS_TYPE=="s"): Mul_name = 'Mul_78'

我现在想给最新的ppyolo+的s模型加NMS,ONNX最后一层变成了tmp_84,修改源码后会报错,那么如果我想给pptoloe+加NMS应该从哪一层开始添加呢?

freecarving avatar Dec 02 '22 06:12 freecarving

是不是gs的api变了,gs_graph.nodes现实的好像不是算子名字

freecarving avatar Dec 02 '22 07:12 freecarving

搞定了谢谢 是paddle2onnx版本不一致的问题,名字乱了,如果不想改代码的话要用0.9版本

freecarving avatar Dec 02 '22 08:12 freecarving

你好,我的tensorrt是7.2版本的,没有EfficientNMS_trt这个算子,那么怎么才能测试我的onnx转换的engine模型呢?也就是初始的onnx用来转换了engine,没有走添加EfficientNMS_trt这一步,因为trt版本里没有这个算子,加了会导致转换engine时报错。

lgy2830 avatar Dec 21 '22 03:12 lgy2830

这个仓库只支持tensorrt8,tensorrt7用tensorrtx的仓库就可以。

Monday-Leo avatar Dec 21 '22 07:12 Monday-Leo