目录

1、出现KeyError

2、解决方法：

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

建议查看博客：

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

1、出现KeyError

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

2、解决方法：

卸载安装的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环境

建议查看博客：

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

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

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