search

GKalliatakis / Keras-VGG16-places365

Keras code and weights files for the VGG16-places365 and VGG16-hybrid1365 CNNs for scene classification
MIT License
194 stars 76 forks source link

Fine Tuning the model #17

Closed ZoulfikarShmayssani closed 3 years ago

ZoulfikarShmayssani commented 3 years ago

Hello.

How do I fine tune the model in order to add more categories. For example I want to add a category called "hypermarkets".

Thank you in advance.

GKalliatakis commented 3 years ago

Hi,

VGG16-places365 models can be used like every other pretrained Keras model (VGG16, ResNet etc.) In order to add your own categories you need to follow the typical transfer-learning workflow.

Hope it helps