mikeyEcology / MLWIC2

Classify camera trap images using machine learning with R Shiny Apps
36 stars 17 forks source link

Initialising with pre-trained weights (transfer learning) #22

Open JoejynWan opened 3 years ago

JoejynWan commented 3 years ago

Hi @mikeyEcology,

With reference to training our own model, I would like to ask if it is possible to use transfer learning and initialise the model using pre-trained weights (e.g. from a model trained with ImageNet), rather than training from randomly initialised weights. I have tried doing this using retrain_from, but since the number of outputs are different, it does not work. Also wondering if the argument transfer_mode has anything to do with this.

Thank you.

Best regards, Joejyn