请问，这个模型可以被转成tensorrt格式吗？

xialuxi / yolov5-car-plate

基于yolov5的车牌检测，包含车牌角点检测

GNU General Public License v3.0

154 stars 40 forks source link

Open HUI11126 opened 3 years ago

HUI11126 commented 3 years ago

onnx -> tensorrt

HUI11126 commented 3 years ago

https://github.com/deepcam-cn/yolov5-face/tree/master/torch2tensorrt 上面这个链接有人把onnx模型转成tensorrt了，但是我把last.pt转成last.onnx之后，再转成last.trt做预测时报了shape不匹配，楼主能否帮忙看一下？谢谢问题链接：https://github.com/deepcam-cn/yolov5-face/pull/50

cnyvfang commented 2 years ago

蹲一个解答

yichuxue commented 2 years ago

你好，你解决这个问题了吗？