PPYOLOE_Tensorrt
PPYOLOE_Tensorrt copied to clipboard
EfficientNMS问题
请教一下作者,EfficientNMS_trt代码中NMS是加在 if(WEIGHTS_TYPE=="s"): Mul_name = 'Mul_78'
我现在想给最新的ppyolo+的s模型加NMS,ONNX最后一层变成了tmp_84,修改源码后会报错,那么如果我想给pptoloe+加NMS应该从哪一层开始添加呢?
是不是gs的api变了,gs_graph.nodes现实的好像不是算子名字
搞定了谢谢 是paddle2onnx版本不一致的问题,名字乱了,如果不想改代码的话要用0.9版本
你好,我的tensorrt是7.2版本的,没有EfficientNMS_trt这个算子,那么怎么才能测试我的onnx转换的engine模型呢?也就是初始的onnx用来转换了engine,没有走添加EfficientNMS_trt这一步,因为trt版本里没有这个算子,加了会导致转换engine时报错。
这个仓库只支持tensorrt8,tensorrt7用tensorrtx的仓库就可以。