项目场景he 问题描述
提示:这里简述项目相关背景:
model = YOLO(model=r'./yolov8m-cls.pt')
加载预训练模型时报错。
ModuleNotFoundError: No module named 'ultralytics.utils'
warning:
bug:
原因分析:
很可能是提前下载的预训练模型出了问题。在运行模型加载代码时有警告。WARNING yolov8m-cls.pt appears to require 'ultralytics.utils', which is not in ultralytics requirements.
,随之抛出bug:ModuleNotFoundError: No module named 'ultralytics.utils'
,同时,During handling of the above exception, another exception occurred:
。Recommend fixes are to train a new model using the latest ‘ultralytics’ package.建议使用最新ultralytics修复模型。
解决方案:
model = YOLO('yolov8n-cls.yaml').load('yolov8n-cls.pt')
使用yolov8n-cls.yaml
,在线下载最新预训练模型,再加载该模型.load('yolov8n-cls.pt')
。
测试加载预训练模型,并训练。运行正常。