gudovskiy / yoloNCS

YOLO object detector for Movidius Neural Compute Stick (NCS)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How are the pretrained weights generated? How do I generate them?

codingquark opened this issue · comments

Are the pretrained weights generated by using create_yolo_caffemodel.py?

If I want to train the model on my own what would the process be? Should I first train yolo by darknet and then use create_yolo_caffemodel.py and create_yolo_prototxt.py?

commented

@codingquark I used pretrained weights. You can either train using darknet and then convert with the scripts or train in caffe directly.

Training in caffe sounds the easiest approach, since I can avoid having to convert to caffe. I have been playing with https://github.com/xingwangsfu/caffe-yolo and it is not working for many .cfgs.