"!yolo task=detect mode=train model=yolov5l6u.pt data={dataset.location}/data.yaml epochs=100 imgsz=640" not running any epochs

abdullahtarek / tennis_analysis

This project analyzes Tennis players in a video to measure their speed, ball shot speed and number of shots. This project will detect players and the tennis ball using YOLO and also utilizes CNNs to extract court keypoints. This hands on project is perfect for polishing your machine learning, and computer vision skills.

459 stars 149 forks source link

"!yolo task=detect mode=train model=yolov5l6u.pt data={dataset.location}/data.yaml epochs=100 imgsz=640" not running any epochs #11

Open jeffnguyen27 opened 5 months ago

jeffnguyen27 commented 5 months ago

The above command stopped right after setup, before running any epoch. I also tried running with yolov5x.pt and yolov516.pt, with no difference. I'm running this on colab. Any idea how to fix this? Thanks.

screen1

Agares9 commented 4 months ago

use GPU

Khush1593 commented 3 months ago

!python train.py --img 640 --batch 10 --epochs 100 --data /content/tennis-ball-detection-6/data.yaml --weights /content/yolov5l6.pt --name yolov5l6_finetune

Also You can use this code for training. This is python code so it works well in CPU as well as in GPU. Here you have to change the value of variables according to your requirements.