赞
踩
目录
若不能解决,建议重新配置YOLOv8环境
建议查看博客:
复现YOLOv8---方法2:推荐-CSDN博客 该博客肯定能为你解决此困扰!!!!!
无论是添加注意力机制或者修改网络,都会出现这个问题,这里只是说明会出现类似的错误
卸载安装的ultralytics包
pip uninstall ultralytics
在mian.py 文件夹下,添加下面代码:ultralytics的绝对路径
- import sys
- sys.path.append("autodl-tmp/yolov8/")
意思为:使用的实时的ultralytics包,而不是之前下载好的本地的包
代码如下:
- import sys
- sys.path.append("autodl-tmp/yolov8/")
-
- from ultralytics import YOLO
-
- if __name__ == '__main__':
- # Load a model
- #直接使用预训练模型创建模型
- # model = YOLO('yolov8n.pt')
- # model.train(**{'cfg':'ultralytics/cfg/default.yaml', 'data':'ultralytics/models/yolo/detect/mydata/traffic.yaml'}, epochs=10, imgsz=640, batch=32)
-
- # #使用yaml配置文件来创建模型,并导入预训练权重
- # model = YOLO('ultralytics/cfg/models/v8/yolov8n-CA.yaml') # build a new model from YAML
- # model.load('yolov8n.pt')
- # model.train(**{'cfg':'ultralytics/cfg/default.yaml', 'data':'ultralytics/models/yolo/detect/mydata/traffic.yaml'}, epochs=10, imgsz=640, batch=32, name='train-ContextAggregation') #name=train-ContextAggregation 更改文件夹的名字
-
- # # # 模型验证:用验证集
- # model = YOLO('runs/detect/train-ContextAggregation/weights/best.pt')
- # model.val(**{'data':'ultralytics/models/yolo/detect/mydata/traffic.yaml', 'name':'val-ContextAggregation', 'batch':32}) #模型验证用验证集
- # model.val(**{'data':'ultralytics/models/yolo/detect/mydata/traffic.yaml', 'split':'test', 'iou':0.9}) #模型验证用测试集
-
- # # 模型推理:
- model = YOLO('runs/detect/train-ContextAggregation/weights/best.pt')
- model.predict(source='ultralytics/assets', name='predict-ContextAggregation', **{'save':True}) #若没有save:True,则会找不到保存路径
应该就可以解决了!!!
若不能解决,建议重新配置YOLOv8环境
建议查看博客:
该博客肯定能为你解决此困扰!!!!!
加油!!!每天学习一点点!!!!
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。