解决ultralytics中的YOLOv8在执行task.py文件添加模块操作出现的KeyError报错

MickeyCV

已于 2024-11-28 22:49:01 修改

阅读量896

点赞数 4

CC 4.0 BY-SA版权

分类专栏：目标检测文章标签： YOLO 深度学习人工智能 python ultralytics bug

于 2024-10-16 00:31:44 首次发布

本文链接：https://blog.csdn.net/weixin_65552509/article/details/142961549

目标检测专栏收录该内容

23 篇文章

订阅专栏

报错详情：

在ultralytics项目文件夹内运行/home/xxx/ultralytics/train.py进行单GPU训练的时候训练可以正常进行

from ultralytics import YOLO

# Load a model
model = YOLO("/home/xxx/ultralytics/ultralytics/cfg/models/v8/yolov8s-FASFF.yaml")  # 

# Train the model
results = model.train(data="/home/xxx/ultralytics/ultralytics/cfg/datasets/VisDrone.yaml", device='1', batch=5, epochs=200, imgsz=640)

但是启动多GPU后训练就会发生KeyEroor报错

# Train the model
results = model.train(data="/home/xxx/ultralytics/ultralytics/cfg/datasets/VisDrone.yaml", device='0,1,2', batch=15, epochs=200, imgsz=640)