search

pochih / FCN-pytorch

🚘 Easiest Fully Convolutional Networks
404 stars 143 forks source link

loading pretrained weights #27

Open soans1994 opened 2 years ago

soans1994 commented 2 years ago

hello author,

i have some queries regarding your implementation.

  1. why did you use the deconv operation 5 times in fcn32, slowly upsampling gives better resultsd than a single conv2d operation?
  2. also, can you please give me some reference how to study about using pretrained weights for custom model. your usage of pretrained weights is good. I want to learn. Can you please give me some inputs. I built a simple fcn model and used pretrained models as follows:

model = fcn() model.load_state_dict(model_zoo.load_url(model_urls['resnet50']), strict=False)

it gives better result, but i am not sure how it uses the weights. whether based on layer names or automatic concat weights based on kernel size.

thank you