tensorrtx
tensorrtx copied to clipboard
yolov8 转wts文件(自定义数据集)失败
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进行训练得到的,报错信息如下
但是,我试着转官方提供的yolov8n.pt时却能正确转换wts文件,不知道为啥会有这种问题....
自定义数据集的best.pt模型是我正常训练yolov8s得到的,之前搞其他C++部署时,转onnx模型是可以正常使用的。
补充一句:查过说是要pip install这个模块,但是我之前训练、验证、测试过程中是正常的,没有出现过提示找不到ultralytics模块,模快更新该模块也无法解决上述问题。