如何解决 Yolo5 中 train.py 程序中的错误

问题描述

我在 GPU 版本的 Google Colab 中运行以下命令

!python train.py --img 640 --batch 4 --epochs 30 \
  --data ./data/clothing.yaml --cfg ./models/yolov5m.yaml --weights yolov5x.pt \
  --name yolov5_clothing --cache

我收到以下错误

Traceback (most recent call last):
  File "train.py",line 404,in <module>
    train(hyp)
  File "train.py",line 70,in train
    data_dict = yaml.load(f,Loader=yaml.FullLoader)  # model dict
AttributeError: module 'yaml' has no attribute 'FullLoader'

我尝试过不同的 yaml 文件，例如 yolol.yaml yolox.yml，但每次都出现此错误。

你能帮我解决这个问题吗？

谢谢

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）

python-3.x yolov5