tensorRT_Pro 直接下载的yolov5的模型，不能转换？必须修改batch后才能转？

直接下载的yolov5的模型，不能转换？必须修改batch后才能转？

Open xinsuinizhuan opened this issue 3 years ago • 5 comments

Feb 21 '22 06:02 xinsuinizhuan

可不可以，直接下载的yolov5模型，onnx转换？要修改yolov5的代码，很麻烦。

Feb 23 '22 06:02 xinsuinizhuan

就类似https://github.com/wang-xinyu/tensorrtx大神的一样

Feb 23 '22 06:02 xinsuinizhuan

可以做到，使用咱这边转换的模型，然后使用[https://github.com/wang-xinyu/tensorrtx中的推理来执行吗？我试了下，执行检查不出目标？

Feb 23 '22 07:02 xinsuinizhuan

我试过了，直接下载的yolov5的预训练模型，batch为1时，可以直接推理；是不是batch非1时，就必须修改导出代码了？

Feb 23 '22 09:02 xinsuinizhuan

如果修改代码都嫌麻烦，那写代码岂不是更麻烦？

Mar 07 '22 03:03 shouxieai