- Download with data from the link below
https://drive.google.com/file/d/1cbbm6_VCCQIHcntppgHIOeGqubedAPVK/view
- Clone the repository from the link below
git clone https://github.com/ultralytics/yolov5
- train
python train.py --img 640 --batch 16 --epochs 3 --data pill.yaml --weights yolov5s.pt
- pretrained model yolov5-small
https://drive.google.com/file/d/1TSwmxqUY9mxTHPdC4XdbnqoTz00Jr-i4/view
- Utilize using torch.hub