ayooshkathuria / YOLO_v3_tutorial_from_scratch

Accompanying code for Paperspace tutorial series "How to Implement YOLO v3 Object Detector from Scratch"
https://blog.paperspace.com/how-to-implement-a-yolo-object-detector-in-pytorch/
2.32k stars 725 forks source link

Training the model? #73

Open Spencer266 opened 1 month ago

Spencer266 commented 1 month ago

Hey I've successfully run the model for object detection, but I'm wondering if this repo only use pre-trained model weights? Is there any part of this repo that was written to train the model or maybe using custom dataset, or do I have to do this myself?. Thanks in advance