tensorRT_Pro icon indicating copy to clipboard operation
tensorRT_Pro copied to clipboard

Published 20 hours ago • verified publisher icon shouxieai

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

Open sdcvs2020 opened this issue 2 years ago • 5 comments

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

sdcvs2020 avatar Nov 26 '22 13:11 sdcvs2020

截图看看？

shouxieai avatar Nov 28 '22 12:11 shouxieai

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

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

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

pzr2508 avatar Nov 28 '22 15:11 pzr2508

截图看看？

寻求大佬！

pzr2508 avatar Nov 28 '22 15:11 pzr2508

大佬你好，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…

lwh1229 avatar Dec 01 '23 06:12 lwh1229