Skip to content

Basic Training Script for YOLOs Model

Pre-release
Pre-release
Compare
Choose a tag to compare
@pchandra90 pchandra90 released this 14 Dec 12:51
· 19 commits to main since this release
  • A single script can be used to train YOLOv5, YOLOv6, and YOLOv7
  • Internally, it downloads code and trained weights from the corresponding repository (yolov5, YOLOv6, or yolov7) for fine-tuning.
  • A sample data in YOLOv5 format is attached.
  • YOLOv6 and YOLOv7 use slightly different formats, but for using this repository, data needs to be in YOLOv5 format.