-
Notifications
You must be signed in to change notification settings - Fork 1.7k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
yolov8 转wts文件(自定义数据集)失败 #1464
Comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
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模块,模快更新该模块也无法解决上述问题。
The text was updated successfully, but these errors were encountered: