How can I retrain this Yolo model for transfer learning?

gudovskiy / yoloNCS

YOLO object detector for Movidius Neural Compute Stick (NCS)

MIT License

180 stars 77 forks source link

How can I retrain this Yolo model for transfer learning? #15

Open WuXinyang2012 opened 6 years ago

WuXinyang2012 commented 6 years ago

Hi I want to use this TinyYolo model but want it only for some other specified categories. Can you give me some instructions on how can I retrain and compile it?

gudovskiy commented 6 years ago

@WuXinyang2012, you can take a look at the original Caffe YOLO repo. https://github.com/xingwangsfu/caffe-yolo or other re implementations.