Skip to content

lzh111222334/yolov5_transformer

 
 

Repository files navigation

yolov5_transformer

yolov5 applying transformer layer

how to train the detector

python train.py --data data.yaml --cfg config.yaml --weights '' --batch-size 64

how to run the detector

python detect.py --source video.mp4 --weights weights.pt --conf 0.25

how to run the tracker

python track.py --source video.mp4 --weights weights.pt --conf 0.25

About

Yolov5 with transformers

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 96.2%
  • Shell 2.8%
  • Dockerfile 1.0%