Skip to content

Latest commit

 

History

History
27 lines (18 loc) · 505 Bytes

Object_detection_README.md

File metadata and controls

27 lines (18 loc) · 505 Bytes

🤔 How to start?

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