K210_Yolo_framework icon indicating copy to clipboard operation
K210_Yolo_framework copied to clipboard

Published 20 hours ago • verified publisher icon zhen8838

pretrained backbone

Open mirecta opened this issue 4 years ago • 1 comments

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

mirecta avatar Mar 13 '20 22:03 mirecta

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

zhen8838 avatar Mar 18 '20 06:03 zhen8838