play_with_tensorrt icon indicating copy to clipboard operation
play_with_tensorrt copied to clipboard
verified publisher icon iwatake2222

请问ufld_v2中模型文件.pth转换到.onnx的方法

Open lwplw opened this issue 3 years ago • 2 comments

Environment (Hardware)

  • Ubuntu 20.04
  • TensorRT-8.4.1.5
  • onnx-1.13.0

Project Name

pj_tensorrt_lane_ultra-fast-lane-detection_v2

Issue Details

使用提供的onnx文件可以成功转到tensorrt且可以运行，但是效果不太理想，尤其是ufldv2_tusimple_res18_320x800.onnx和ufldv2_tusimple_res34_320x800.onnx，我想要尝试将自己训练的.pth转到tensorrt，请问该方法中.pth转换到.onnx是怎么实现的，有相关代码吗？

lwplw avatar Mar 14 '23 08:03 lwplw

请问你解决了没。 网上一般的pt->onnx的方法是不能用吗

FunnyWii avatar Feb 20 '24 01:02 FunnyWii

pth转onnx很简单吧，原项目Coex转onnx挺顺利的，就是从onnx来build TRT engine不成功，里面有TRT未定义的op，我就没有再细究了

Cai-RS avatar Apr 10 '24 21:04 Cai-RS