search

titu1994 / DenseNet

DenseNet implementation in Keras
MIT License
707 stars 294 forks source link

How can I retrain this model ? #18

Closed beerboaa closed 7 years ago

beerboaa commented 7 years ago

Thank you for this implementation. I want to know how to use this as transfer learning. Would you please tell me step by step.

titu1994 commented 7 years ago

Refer to Keras fine tuning blog post. You can swap out the VGG model there for DenseNet from here, but since I don't have weights for imagenet, you will have to use 'cifar10' instead of 'imagenet'

titu1994 commented 7 years ago

This is what I meant - https://blog.keras.io/building-powerful-image-classification-models-using-very-little-data.html

beerboaa commented 7 years ago

Thank you