Skip to content

pat-langevin/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

Contributors 2

  •  
  •