在 YOLOV5 中运行 train.py 时出现属性错误谁能帮我这个?

问题描述

当我在我的系统中为 YOLO V5 运行 python train.py --img 640 --batch 4 --epochs 5 --data training/dataset.yaml --cfg training/yolov5l.yaml --weights yolov5l.pt我得到以下信息和原因:

回溯(最近一次调用最后一次): 文件“train.py”,第 544 行,在 火车(hyp,选择,设备,tb_writer) 文件“train.py”,第 72 行,在火车中 wandb_logger = WandbLogger(opt,save_dir.stem,run_id,data_dict) init 中的文件“D:\sandra\ai.projects\yolo\yolov5\utils\wandb_logging\wandb_utils.py”,第 108 行 self.data_dict = self.setup_training(opt,data_dict) 文件“D:\sandra\ai.projects\yolo\yolov5\utils\wandb_logging\wandb_utils.py”,第 139 行,在 setup_training self.train_artifact_path,self.train_artifact = self.download_dataset_artifact(data_dict.get('train'),AttributeError: 'str' 对象没有属性 'get'

wandb:等待 W&B 进程完成,PID 22204 wandb:程序失败,代码为 1。 wandb:在以下位置查找此运行的用户日志:D:\sandra\ai.projects\yolo\yolov5\wandb\offline-run-20210427_130128-jr2z73rr\logs\debug.log wandb:在以下位置查找此运行的内部日志:D:\sandra\ai.projects\yolo\yolov5\wandb\offline-run-20210427_130128-jr2z73rr\logs\debug-internal.log wandb:您可以通过运行以下命令将此运行同步到云: wandb: wandb 同步 D:\sandra\ai.projects\yolo\yolov5\wandb\offline-run-20210427_130128-jr2z73rr

解决方法

查看错误回溯,您的数据集配置文件似乎缺少一些信息,或者很可能格式错误。该文件应包含训练和验证集路径。举个例子:

train: path/to/train # Notice the spaces
val: path/to/val
...