search

zhen8838 / K210_Yolo_framework

Yolo v3 framework base on tensorflow, support multiple models, multiple datasets, any number of output layers, any number of anchors, model prune, and portable model to K210 !
MIT License
248 stars 73 forks source link

pretrained backbone #9

Open mirecta opened 4 years ago

mirecta commented 4 years ago

Hi can i use pretrained backbone mobilenet darknet network as feature extractor, freeze this layers and train only detector layers ? AKA transfer learning ?

zhen8838 commented 4 years ago

yes , you can. but mobilenet backbone for k210 need to modify. you can refer my code https://github.com/zhen8838/K210_Yolo_framework/blob/master/models/keras_mobilenet.py