search

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 724 forks source link

How can we train the model #12

Open lzxzy opened 6 years ago

lzxzy commented 6 years ago

First thanks a lot for you wrote the tutorial, which helps me understand the yolo algorithm simply.So the next, if I want to train the model by myself, rely on the work you have done, how can I do this? Can you give me some tips please, which mean how can I implement the backward process?

ayooshkathuria commented 6 years ago

I'm gonna write the train code very soon, and write another tutorial. Been really busy with this month with my college ending, moving stuff back home and looking for jobs.

KasunSandaras commented 5 years ago

@ayooshkathuria thank you very much for your great tutorial series. Can we have a tutorial series to train our own model and use with this code. At least can i have little guide on that.thank you very much.

glenn-jocher commented 5 years ago

PyTorch YOLOv3 training is as easy as this in https://github.com/ultralytics/yolov3. You can even run this directly in our Colab Notebook.

git clone https://github.com/ultralytics/yolov3  # clone
bash yolov3/data/get_coco_dataset_gdrive.sh  # copy COCO2014 dataset (20GB)
cd yolov3
python3 train.py  # train :)