YOLO_v3_tutorial_from_scratch icon indicating copy to clipboard operation
YOLO_v3_tutorial_from_scratch copied to clipboard

How can we train the model

Open lzxzy opened this issue 6 years ago • 3 comments

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?

lzxzy avatar May 04 '18 03:05 lzxzy

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.

ayooshkathuria avatar May 18 '18 10:05 ayooshkathuria

@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.

KasunSandaras avatar Dec 23 '18 14:12 KasunSandaras

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 :)

glenn-jocher avatar Aug 03 '19 14:08 glenn-jocher