shouxieai/tensorRT_Pro

大佬你好，在转换YOLOV7为TRT模型时遇到错误

sdcvs2020 opened this issue · 5 comments

sdcvs2020 commented

在转换YOLOV7为TRT模型时遇到错误Could not found the input/output node 'output', please makesure your model，但尝试用TRTEXEC可以转换，请问怎么解决

shouxieai commented

截图看看？

pzr2508 commented

大佬你好，yolov5为RTR模型也会存在这个问题，导出的onnx模型是有输入“images”节点和输出“output”节点，不知道“output0”节点名称从哪里来的？
984aa11aa35a2b73421188f970cc670
3fc7eab119ac9461488b143b9e439d3
我只有将trt_infer.cpp中blobsNameMapper_字典key “output0” 强制替换为 “output” 就可以运行。
72e10197121b2d9b6d741464240c7dc

不知道大佬有没有其他较好的方法？

#################################################
###############################################
###########问题已经解决
需要将workspace下的之前编译过生成xxx.txt的文件删除，如果不删除就不会重新编译，output名称为上一次trtmodel的名称，
所以可能会与当前任务的output 名称不一致。

pzr2508 commented

截图看看？

寻求大佬！

lwh1229 commented

大佬你好，yolov5为RTR模型也会存在这个问题，导出的onnx模型是有输入“images”节点和输出“output”节点，不知道“output0”节点名称从哪里来的？ 984aa11aa35a2b73421188f970cc670 3fc7eab119ac9461488b143b9e439d3 我只有将trt_infer.cpp中blobsNameMapper_字典key “output0” 强制替换为 “output” 就可以运行。 72e10197121b2d9b6d741464240c7dc

不知道大佬有没有其他较好的方法？

################################################# ############################################### ###########问题已经解决 需要将workspace下的之前编译过生成xxx.txt的文件删除，如果不删除就不会重新编译，output名称为上一次trtmodel的名称， 所以可能会与当前任务的output 名称不一致。

大佬，为什么我删了txt也报错，像你那样改代码又报了另一个错
Uploading nvidia.png…