For CPU
! git clone https://github.com/ansharora7/vehicle_count_colab.git
% cd /content/vehicle_count_colab/Yolov5_DeepSort_Pytorch
! pip install -r requirements.txt
! sh run.sh
For GPU
! git clone https://github.com/ansharora7/vehicle_count_colab.git
% cd /content/vehicle_count_colab/Yolov5_DeepSort_Pytorch
! pip install -r requirements.txt
! sh run_gpu.sh
To check more pre-trained YOLOv5 models, https://github.com/ultralytics/yolov5/wiki/Tips-for-Best-Training-Results & https://github.com/ultralytics/yolov5#pretrained-checkpoints