pytorch-yolo-v3 icon indicating copy to clipboard operation
pytorch-yolo-v3 copied to clipboard

Is there any plan to add the training part?

Open Sephirot1st opened this issue 6 years ago • 6 comments

Sephirot1st avatar Dec 12 '18 09:12 Sephirot1st

That would be very helpful =)

TimDNN avatar Dec 12 '18 12:12 TimDNN

https://pjreddie.com/darknet/yolo/

BryanBao avatar Jan 05 '19 07:01 BryanBao

its mean, can i train by darknet,and using by the code?

apple1987 avatar Jan 24 '19 06:01 apple1987

I've successfully trained YOLOv3 with PyTorch from scratch using this repo. It also does automatic image augmentation, and approaches the darknet mAP after about 65 epochs. https://github.com/ultralytics/yolov3

fourth-archive avatar Feb 09 '19 18:02 fourth-archive

I've successfully trained YOLOv3 with PyTorch from scratch using this repo. It also does automatic image augmentation, and approaches the darknet mAP after about 65 epochs. https://github.com/ultralytics/yolov3

How did you do it? That is what I exactly want. I want to use the YOLOv3 with PyTorch from scratch as the core of the code with the parts from the directory you linked which are necessary for training. Im not sure which parts should I copy besides the training.py, and which parts should I overwrite with in the YOLO from scratch with code from the ultralytics directory. Thanks for the help in advance.

emmbertelen avatar Aug 01 '19 11:08 emmbertelen

@emmbertelen starting training is as easy as this in https://github.com/ultralytics/yolov3. You can even run this directly in our Colab Notebook: https://github.com/ultralytics/yolov3/blob/master/examples.ipynb

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