当前位置:   article > 正文

复现YOLOv8所遇到的问题:KeyError_yolov8 keyerror: 'backbone

yolov8 keyerror: 'backbone

目录

1、出现KeyError

2、解决方法:


若不能解决,建议重新配置YOLOv8环境

建议查看博客:

复现YOLOv8---方法2:推荐-CSDN博客       该博客肯定能为你解决此困扰!!!!!

1、出现KeyError

无论是添加注意力机制或者修改网络,都会出现这个问题,这里只是说明会出现类似的错误

2、解决方法:

卸载安装的ultralytics包

pip uninstall ultralytics

在mian.py 文件夹下,添加下面代码:ultralytics的绝对路径

  1. import sys
  2. sys.path.append("autodl-tmp/yolov8/")

 意思为:使用的实时的ultralytics包,而不是之前下载好的本地的包

代码如下:

  1. import sys
  2. sys.path.append("autodl-tmp/yolov8/")
  3. from ultralytics import YOLO
  4. if __name__ == '__main__':
  5. # Load a model
  6. #直接使用预训练模型创建模型
  7. # model = YOLO('yolov8n.pt')
  8. # model.train(**{'cfg':'ultralytics/cfg/default.yaml', 'data':'ultralytics/models/yolo/detect/mydata/traffic.yaml'}, epochs=10, imgsz=640, batch=32)
  9. # #使用yaml配置文件来创建模型,并导入预训练权重
  10. # model = YOLO('ultralytics/cfg/models/v8/yolov8n-CA.yaml') # build a new model from YAML
  11. # model.load('yolov8n.pt')
  12. # 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 更改文件夹的名字
  13. # # # 模型验证:用验证集
  14. # model = YOLO('runs/detect/train-ContextAggregation/weights/best.pt')
  15. # model.val(**{'data':'ultralytics/models/yolo/detect/mydata/traffic.yaml', 'name':'val-ContextAggregation', 'batch':32}) #模型验证用验证集
  16. # model.val(**{'data':'ultralytics/models/yolo/detect/mydata/traffic.yaml', 'split':'test', 'iou':0.9}) #模型验证用测试集
  17. # # 模型推理:
  18. model = YOLO('runs/detect/train-ContextAggregation/weights/best.pt')
  19. model.predict(source='ultralytics/assets', name='predict-ContextAggregation', **{'save':True}) #若没有save:True,则会找不到保存路径

 应该就可以解决了!!!

若不能解决,建议重新配置YOLOv8环境

建议查看博客:

复现YOLOv8---方法2:推荐-CSDN博客

 该博客肯定能为你解决此困扰!!!!!


加油!!!每天学习一点点!!!!

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Monodyee/article/detail/357073
推荐阅读
相关标签
  

闽ICP备14008679号