yolov5x train.py error

[paragjp]

Hi

I am trying to run yolov5x module on google colab on GPU, I have successfully imported following

!pip install torch
!!pip install PyYAML
!!pip install git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI

from pathlib import Path
from tqdm import tqdm
import numpy as np
import json
import urllib
import PIL.Image as Image
import cv2
import torch
import torchvision
from IPython.display import display
from sklearn.model_selection import train_test_split
import seaborn as sns
from pylab import rcParams
import matplotlib.pyplot as plt
from matplotlib import rc
%matplotlib inline
%config InlineBackend.figure_format='retina'

When I am running following command
!python train.py --img 640 --batch 4 --epochs 30
--data ./data/clothing.yaml --cfg ./models/yolov5x.yaml --weights yolov5x.pt
--name yolov5x_clothing --cache

It is giving me following error

Traceback (most recent call last):
File "train.py", line 404, in
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'

Pl help

[github-actions]

👋 Hello @paragjp, thank you for your interest in 🚀 YOLOv5! Please visit our ⭐️ Tutorials to get started, where you can find quickstart guides for simple tasks like Custom Data Training all the way to advanced concepts like Hyperparameter Evolution.

If this is a 🐛 Bug Report, please provide screenshots and minimum viable code to reproduce your issue, otherwise we can not help you.

If this is a custom training ❓ Question, please provide as much information as possible, including dataset images, training logs, screenshots, and a public link to online W&B logging if available.

For business inquiries or professional support requests please visit https://www.ultralytics.com or email Glenn Jocher at glenn.jocher@ultralytics.com.

Requirements

Python 3.8 or later with all requirements.txt dependencies installed, including torch>=1.7. To install run:

$ pip install -r requirements.txt

Environments

YOLOv5 may be run in any of the following up-to-date verified environments (with all dependencies including CUDA/CUDNN, Python and PyTorch preinstalled):

• Google Colab and Kaggle notebooks with free GPU:
• Google Cloud Deep Learning VM. See GCP Quickstart Guide
• Amazon Deep Learning AMI. See AWS Quickstart Guide
• Docker Image. See Docker Quickstart Guide

Status

If this badge is green, all YOLOv5 GitHub Actions Continuous Integration (CI) tests are currently passing. CI tests verify correct operation of YOLOv5 training (train.py), testing (test.py), inference (detect.py) and export (export.py) on MacOS, Windows, and Ubuntu every 24 hours and on every commit.

[paragjp]

Hi

Pl find enclosed screen shot of an error

I am using Kaggle clothing dataset.

Thanks

[glenn-jocher]

@paragjp your pyyaml is not meeting requirements in requirements.txt.

Requirements

Python 3.8 or later with all requirements.txt dependencies installed, including torch>=1.7. To install run:

$ pip install -r requirements.txt

Environments

YOLOv5 may be run in any of the following up-to-date verified environments (with all dependencies including CUDA/CUDNN, Python and PyTorch preinstalled):

• Google Colab and Kaggle notebooks with free GPU:
• Google Cloud Deep Learning VM. See GCP Quickstart Guide
• Amazon Deep Learning AMI. See AWS Quickstart Guide
• Docker Image. See Docker Quickstart Guide

Status

If this badge is green, all YOLOv5 GitHub Actions Continuous Integration (CI) tests are passing. These tests evaluate proper operation of basic YOLOv5 functionality, including training (train.py), testing (test.py), inference (detect.py) and export (export.py) on MacOS, Windows, and Ubuntu.

[paragjp]

@glenn-jocher

Thanks after installing PyYAML to 5.3.1 this issue has been resolved now I am facing following issue

Traceback (most recent call last):
File "train.py", line 404, in
train(hyp)
File "train.py", line 122, in train
raise KeyError(s) from e
KeyError: "yolov5x.pt is not compatible with /content/yolov5/models/yolov5x.yaml. Specify --weights '' or specify a --cfg compatible with yolov5x.pt.

My command is

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

Pl help, Thanks

[paragjp]

Hi

Any updates ?

Thanks

[NanoCode012]

@paragjp, I think it's a mismatch in code and model weights. You should pull the latest code and delete weights, so they can be auto-downloaded again.

[paragjp]

@NanoCode012

Thanks it is working now