earhian / Humpback-Whale-Identification-1st-

https://www.kaggle.com/c/humpback-whale-identification
641 stars 184 forks source link

'senet154' pretrained model needs input_channel=3, but your model's input is 4. and I can't find the transformation between the input channel.How you do that ? #6

Closed Bigwode closed 5 years ago

Bigwode commented 5 years ago

'senet154' model pretrained on imagenet needs input_channel=3, but your model's input is 4. And I can't find the transformation between the input channel.How you do that ?Thanks!

earhian commented 5 years ago

I used try-except for loading pretrained model in line 367 of senet.py.

Bigwode commented 5 years ago

@earhian Oww, It means that the pretrained weights in the first Conv are not loaded in your model, right?

earhian commented 5 years ago

yes!