目录
若不能解决,建议重新配置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环境
建议查看博客:
该博客肯定能为你解决此困扰!!!!!
加油!!!每天学习一点点!!!!