search

e-lab / LinkNet

168 stars 42 forks source link

Fine Tuning #3

Closed MyVanitar closed 7 years ago

MyVanitar commented 7 years ago

Hi,

is there any possibility to fine-tune this model on a custom datase with different number of classes? The pre-trained weights must be exist also, as I know.

codeAC29 commented 7 years ago

@VanitarNordic yes you can fine-tune your model or train it completely from scratch on any custom dataset. You will just have to write a data loader. One of the data loader located here can be used as an example.

MyVanitar commented 7 years ago

Thank you. Two tips remains:

1) The pre-trained weights (CityScapes, CamVid) and where to load them 2) Where to change number of classes

codeAC29 commented 7 years ago
  1. Add a model.lua file similar to this and load your pretrained model instead of defining all the layers.
  2. Just modify your class list. For example modify this line in case of camvid dataset.