tensorrtx icon indicating copy to clipboard operation
tensorrtx copied to clipboard

yolov8 转wts文件(自定义数据集)失败

Open pcycccccc opened this issue 3 months ago • 2 comments

Env

  • GPU: RTX 3080. Env

  • GPU: RTX 3080.

  • OS, Win10.

  • Cuda version 11.8

  • TensorRT version 8.6.1

我在自己训练好的模型转wts的时候失败了,提示找不到模块 No module named 'ultralytics.utils',这个pt模型是我用yolov8s.pt进行训练得到的,报错信息如下 image 但是,我试着转官方提供的yolov8n.pt时却能正确转换wts文件,不知道为啥会有这种问题....

自定义数据集的best.pt模型是我正常训练yolov8s得到的,之前搞其他C++部署时,转onnx模型是可以正常使用的。

补充一句:查过说是要pip install这个模块,但是我之前训练、验证、测试过程中是正常的,没有出现过提示找不到ultralytics模块,模快更新该模块也无法解决上述问题。

pcycccccc avatar Mar 22 '24 03:03 pcycccccc