Loading the model

[GLivshits]

Hello. Im trying to use ur model. I've figured out how to set the appropriate size of an image (it should be a multiple of 16). However, i can't load weights via state dict. U know, all this pytorch shit with missing keys. If I dont load state dict - model outputs some crap.
Any advice on that?

[kampta]

Please share what you've tried so far to load the model and the error you're getting

[dwil2444]

hi @kampta I tried loading the model and got the following error:

RuntimeError: Error(s) in loading state_dict for MobileNetV2_unet:
Missing key(s) in state_dict: "backbone.features.0.0.weight", "backbone.features.0.1.weight", "backbone.features.0.1.bias", "backbone.features.0.1.running_mean", "backbone.features.0.1.running_var", "backbone.features.1.conv.0.weight", "backbone.features.1.conv.1.weight", "backbone.features.1.conv.1.bias", "backbone.features.1.conv.1.running_mean", "backbone.features.1.conv.1.running_var", "backbone.features.1.conv.3.weight", "backbone.features.1.conv.4.weight", "backbone.features.1.conv.4.bias", "backbone.features.1.conv.4.running_mean", "backbone.features.1.conv.4.running_var", "backbone.features.2.conv.0.weight", "backbone.features.2.conv.1.weight", "backbone.features.2.conv.1.bias", "backbone.features.2.conv.1.running_mean", "backbone.features.2.conv.1.running_var", "backbone.features.2.conv.3.weight", "backbone.features.2.conv.4.weight", "backbone.features.2.conv.4.bias", "backbone.features.2.conv.4.running_mean", "backbone.features.2.conv.4.running_var", "backbone.features.2.conv.6.weight", "backbone.features.2.conv.7.weight", "backbone.features.2.conv.7.bias", "backbone.features.2.conv.7.running_mean", "backbone.features.2.conv.7.running_var", "backbone.features.3.conv.0.weight", "backbone.features.3.conv.1.weight", "backbone.features.3.conv.1.bias", "backbone.features.3.conv.1.running_mean", "backbone.features.3.conv.1.running_var", "backbone.features.3.conv.3.weight", "backbone.features.3.conv.4.weight", "backbone.features.3.conv.4.bias", "backbone.features.3.conv.4.running_mean", "backbone.features.3.conv.4.running_var", "backbone.features.3.conv.6.weight", "backbone.features.3.conv.7.weight", "backbone.features.3.conv.7.bias", "backbone.features.3.conv.7.running_mean", "backbone.features.3.conv.7.running_var", "backbone.features.4.conv.0.weight", "backbone.features.4.conv.1.weight", "backbone.features.4.conv.1.bias", "backbone.features.4.conv.1.running_mean", "backbone.features.4.conv.1.running_var", "backbone.features.4.conv.3.weight", "backbone.features.4.conv.4.weight", "backbone.features.4.conv.4.bias", "backbone.features.4.conv.4.running_mean", "backbone.features.4.conv.4.running_var", "backbone.features.4.conv.6.weight", "backbone.features.4.conv.7.weight", "backbone.features.4.conv.7.bias", "backbone.features.4.conv.7.running_mean", "backbone.features.4.conv.7.running_var", "backbone.features.5.conv.0.weight", "backbone.features.5.conv.1.weight", "backbone.features.5.conv.1.bias", "backbone.features.5.conv.1.running_mean", "backbone.features.5.conv.1.running_var", "backbone.features.5.conv.3.weight", "backbone.features.5.conv.4.weight", "backbone.features.5.conv.4.bias", "backbone.features.5.conv.4.running_mean", "backbone.features.5.conv.4.running_var", "backbone.features.5.conv.6.weight", "backbone.features.5.conv.7.weight", "backbone.features.5.conv.7.bias", "backbone.features.5.conv.7.running_mean", "backbone.features.5.conv.7.running_var", "backbone.features.6.conv.0.weight", "backbone.features.6.conv.1.weight", "backbone.features.6.conv.1.bias", "backbone.features.6.conv.1.running_mean", "backbone.features.6.conv.1.running_var", "backbone.features.6.conv.3.weight", "backbone.features.6.conv.4.weight", "backbone.features.6.conv.4.bias", "backbone.features.6.conv.4.running_mean", "backbone.features.6.conv.4.running_var", "backbone.features.6.conv.6.weight", "backbone.features.6.conv.7.weight", "backbone.features.6.conv.7.bias", "backbone.features.6.conv.7.running_mean", "backbone.features.6.conv.7.running_var", "backbone.features.7.conv.0.weight", "backbone.features.7.conv.1.weight", "backbone.features.7.conv.1.bias", "backbone.features.7.conv.1.running_mean", "backbone.features.7.conv.1.running_var", "backbone.features.7.conv.3.weight", "backbone.features.7.conv.4.weight", "backbone.features.7.conv.4.bias", "backbone.features.7.conv.4.running_mean", "backbone.features.7.conv.4.running_var", "backbone.features.7.conv.6.weight", "backbone.features.7.conv.7.weight", "backbone.features.7.conv.7.bias", "backbone.features.7.conv.7.running_mean", "backbone.features.7.conv.7.running_var", "backbone.features.8.conv.0.weight", "backbone.features.8.conv.1.weight", "backbone.features.8.conv.1.bias", "backbone.features.8.conv.1.running_mean", "backbone.features.8.conv.1.running_var", "backbone.features.8.conv.3.weight", "backbone.features.8.conv.4.weight", "backbone.features.8.conv.4.bias", "backbone.features.8.conv.4.running_mean", "backbone.features.8.conv.4.running_var", "backbone.features.8.conv.6.weight", "backbone.features.8.conv.7.weight", "backbone.features.8.conv.7.bias", "backbone.features.8.conv.7.running_mean", "backbone.features.8.conv.7.running_var", "backbone.features.9.conv.0.weight", "backbone.features.9.conv.1.weight", "backbone.features.9.conv.1.bias", "backbone.features.9.conv.1.running_mean", "backbone.features.9.conv.1.running_var", "backbone.features.9.conv.3.weight", "backbone.features.9.conv.4.weight", "backbone.features.9.conv.4.bias", "backbone.features.9.conv.4.running_mean", "backbone.features.9.conv.4.running_var", "backbone.features.9.conv.6.weight", "backbone.features.9.conv.7.weight", "backbone.features.9.conv.7.bias", "backbone.features.9.conv.7.running_mean", "backbone.features.9.conv.7.running_var", "backbone.features.10.conv.0.weight", "backbone.features.10.conv.1.weight", "backbone.features.10.conv.1.bias", "backbone.features.10.conv.1.running_mean", "backbone.features.10.conv.1.running_var", "backbone.features.10.conv.3.weight", "backbone.features.10.conv.4.weight", "backbone.features.10.conv.4.bias", "backbone.features.10.conv.4.running_mean", "backbone.features.10.conv.4.running_var", "backbone.features.10.conv.6.weight", "backbone.features.10.conv.7.weight", "backbone.features.10.conv.7.bias", "backbone.features.10.conv.7.running_mean", "backbone.features.10.conv.7.running_var", "backbone.features.11.conv.0.weight", "backbone.features.11.conv.1.weight", "backbone.features.11.conv.1.bias", "backbone.features.11.conv.1.running_mean", "backbone.features.11.conv.1.running_var", "backbone.features.11.conv.3.weight", "backbone.features.11.conv.4.weight", "backbone.features.11.conv.4.bias", "backbone.features.11.conv.4.running_mean", "backbone.features.11.conv.4.running_var", "backbone.features.11.conv.6.weight", "backbone.features.11.conv.7.weight", "backbone.features.11.conv.7.bias", "backbone.features.11.conv.7.running_mean", "backbone.features.11.conv.7.running_var", "backbone.features.12.conv.0.weight", "backbone.features.12.conv.1.weight", "backbone.features.12.conv.1.bias", "backbone.features.12.conv.1.running_mean", "backbone.features.12.conv.1.running_var", "backbone.features.12.conv.3.weight", "backbone.features.12.conv.4.weight", "backbone.features.12.conv.4.bias", "backbone.features.12.conv.4.running_mean", "backbone.features.12.conv.4.running_var", "backbone.features.12.conv.6.weight", "backbone.features.12.conv.7.weight", "backbone.features.12.conv.7.bias", "backbone.features.12.conv.7.running_mean", "backbone.features.12.conv.7.running_var", "backbone.features.13.conv.0.weight", "backbone.features.13.conv.1.weight", "backbone.features.13.conv.1.bias", "backbone.features.13.conv.1.running_mean", "backbone.features.13.conv.1.running_var", "backbone.features.13.conv.3.weight", "backbone.features.13.conv.4.weight", "backbone.features.13.conv.4.bias", "backbone.features.13.conv.4.running_mean", "backbone.features.13.conv.4.running_var", "backbone.features.13.conv.6.weight", "backbone.features.13.conv.7.weight", "backbone.features.13.conv.7.bias", "backbone.features.13.conv.7.running_mean", "backbone.features.13.conv.7.running_var", "backbone.features.14.conv.0.weight", "backbone.features.14.conv.1.weight", "backbone.features.14.conv.1.bias", "backbone.features.14.conv.1.running_mean", "backbone.features.14.conv.1.running_var", "backbone.features.14.conv.3.weight", "backbone.features.14.conv.4.weight", "backbone.features.14.conv.4.bias", "backbone.features.14.conv.4.running_mean", "backbone.features.14.conv.4.running_var", "backbone.features.14.conv.6.weight", "backbone.features.14.conv.7.weight", "backbone.features.14.conv.7.bias", "backbone.features.14.conv.7.running_mean", "backbone.features.14.conv.7.running_var", "backbone.features.15.conv.0.weight", "backbone.features.15.conv.1.weight", "backbone.features.15.conv.1.bias", "backbone.features.15.conv.1.running_mean", "backbone.features.15.conv.1.running_var", "backbone.features.15.conv.3.weight", "backbone.features.15.conv.4.weight", "backbone.features.15.conv.4.bias", "backbone.features.15.conv.4.running_mean", "backbone.features.15.conv.4.running_var", "backbone.features.15.conv.6.weight", "backbone.features.15.conv.7.weight", "backbone.features.15.conv.7.bias", "backbone.features.15.conv.7.running_mean", "backbone.features.15.conv.7.running_var", "backbone.features.16.conv.0.weight", "backbone.features.16.conv.1.weight", "backbone.features.16.conv.1.bias", "backbone.features.16.conv.1.running_mean", "backbone.features.16.conv.1.running_var", "backbone.features.16.conv.3.weight", "backbone.features.16.conv.4.weight", "backbone.features.16.conv.4.bias", "backbone.features.16.conv.4.running_mean", "backbone.features.16.conv.4.running_var", "backbone.features.16.conv.6.weight", "backbone.features.16.conv.7.weight", "backbone.features.16.conv.7.bias", "backbone.features.16.conv.7.running_mean", "backbone.features.16.conv.7.running_var", "backbone.features.17.conv.0.weight", "backbone.features.17.conv.1.weight", "backbone.features.17.conv.1.bias", "backbone.features.17.conv.1.running_mean", "backbone.features.17.conv.1.running_var", "backbone.features.17.conv.3.weight", "backbone.features.17.conv.4.weight", "backbone.features.17.conv.4.bias", "backbone.features.17.conv.4.running_mean", "backbone.features.17.conv.4.running_var", "backbone.features.17.conv.6.weight", "backbone.features.17.conv.7.weight", "backbone.features.17.conv.7.bias", "backbone.features.17.conv.7.running_mean", "backbone.features.17.conv.7.running_var", "backbone.features.18.0.weight", "backbone.features.18.1.weight", "backbone.features.18.1.bias", "backbone.features.18.1.running_mean", "backbone.features.18.1.running_var", "backbone.classifier.1.weight", "backbone.classifier.1.bias", "dconv1.weight", "dconv1.bias", "invres1.conv.0.weight", "invres1.conv.1.weight", "invres1.conv.1.bias", "invres1.conv.1.running_mean", "invres1.conv.1.running_var", "invres1.conv.3.weight", "invres1.conv.4.weight", "invres1.conv.4.bias", "invres1.conv.4.running_mean", "invres1.conv.4.running_var", "invres1.conv.6.weight", "invres1.conv.7.weight", "invres1.conv.7.bias", "invres1.conv.7.running_mean", "invres1.conv.7.running_var", "dconv2.weight", "dconv2.bias", "invres2.conv.0.weight", "invres2.conv.1.weight", "invres2.conv.1.bias", "invres2.conv.1.running_mean", "invres2.conv.1.running_var", "invres2.conv.3.weight", "invres2.conv.4.weight", "invres2.conv.4.bias", "invres2.conv.4.running_mean", "invres2.conv.4.running_var", "invres2.conv.6.weight", "invres2.conv.7.weight", "invres2.conv.7.bias", "invres2.conv.7.running_mean", "invres2.conv.7.running_var", "dconv3.weight", "dconv3.bias", "invres3.conv.0.weight", "invres3.conv.1.weight", "invres3.conv.1.bias", "invres3.conv.1.running_mean", "invres3.conv.1.running_var", "invres3.conv.3.weight", "invres3.conv.4.weight", "invres3.conv.4.bias", "invres3.conv.4.running_mean", "invres3.conv.4.running_var", "invres3.conv.6.weight", "invres3.conv.7.weight", "invres3.conv.7.bias", "invres3.conv.7.running_mean", "invres3.conv.7.running_var", "dconv4.weight", "dconv4.bias", "invres4.conv.0.weight", "invres4.conv.1.weight", "invres4.conv.1.bias", "invres4.conv.1.running_mean", "invres4.conv.1.running_var", "invres4.conv.3.weight", "invres4.conv.4.weight", "invres4.conv.4.bias", "invres4.conv.4.running_mean", "invres4.conv.4.running_var", "invres4.conv.6.weight", "invres4.conv.7.weight", "invres4.conv.7.bias", "invres4.conv.7.running_mean", "invres4.conv.7.running_var", "conv_last.weight", "conv_last.bias", "conv_score.weight", "conv_score.bias".
Unexpected key(s) in state_dict: "features.0.0.weight", "features.0.1.weight", "features.0.1.bias", "features.0.1.running_mean", "features.0.1.running_var", "features.1.conv.0.weight", "features.1.conv.1.weight", "features.1.conv.1.bias", "features.1.conv.1.running_mean", "features.1.conv.1.running_var", "features.1.conv.3.weight", "features.1.conv.4.weight", "features.1.conv.4.bias", "features.1.conv.4.running_mean", "features.1.conv.4.running_var", "features.2.conv.0.weight", "features.2.conv.1.weight", "features.2.conv.1.bias", "features.2.conv.1.running_mean", "features.2.conv.1.running_var", "features.2.conv.3.weight", "features.2.conv.4.weight", "features.2.conv.4.bias", "features.2.conv.4.running_mean", "features.2.conv.4.running_var", "features.2.conv.6.weight", "features.2.conv.7.weight", "features.2.conv.7.bias", "features.2.conv.7.running_mean", "features.2.conv.7.running_var", "features.3.conv.0.weight", "features.3.conv.1.weight", "features.3.conv.1.bias", "features.3.conv.1.running_mean", "features.3.conv.1.running_var", "features.3.conv.3.weight", "features.3.conv.4.weight", "features.3.conv.4.bias", "features.3.conv.4.running_mean", "features.3.conv.4.running_var", "features.3.conv.6.weight", "features.3.conv.7.weight", "features.3.conv.7.bias", "features.3.conv.7.running_mean", "features.3.conv.7.running_var", "features.4.conv.0.weight", "features.4.conv.1.weight", "features.4.conv.1.bias", "features.4.conv.1.running_mean", "features.4.conv.1.running_var", "features.4.conv.3.weight", "features.4.conv.4.weight", "features.4.conv.4.bias", "features.4.conv.4.running_mean", "features.4.conv.4.running_var", "features.4.conv.6.weight", "features.4.conv.7.weight", "features.4.conv.7.bias", "features.4.conv.7.running_mean", "features.4.conv.7.running_var", "features.5.conv.0.weight", "features.5.conv.1.weight", "features.5.conv.1.bias", "features.5.conv.1.running_mean", "features.5.conv.1.running_var", "features.5.conv.3.weight", "features.5.conv.4.weight", "features.5.conv.4.bias", "features.5.conv.4.running_mean", "features.5.conv.4.running_var", "features.5.conv.6.weight", "features.5.conv.7.weight", "features.5.conv.7.bias", "features.5.conv.7.running_mean", "features.5.conv.7.running_var", "features.6.conv.0.weight", "features.6.conv.1.weight", "features.6.conv.1.bias", "features.6.conv.1.running_mean", "features.6.conv.1.running_var", "features.6.conv.3.weight", "features.6.conv.4.weight", "features.6.conv.4.bias", "features.6.conv.4.running_mean", "features.6.conv.4.running_var", "features.6.conv.6.weight", "features.6.conv.7.weight", "features.6.conv.7.bias", "features.6.conv.7.running_mean", "features.6.conv.7.running_var", "features.7.conv.0.weight", "features.7.conv.1.weight", "features.7.conv.1.bias", "features.7.conv.1.running_mean", "features.7.conv.1.running_var", "features.7.conv.3.weight", "features.7.conv.4.weight", "features.7.conv.4.bias", "features.7.conv.4.running_mean", "features.7.conv.4.running_var", "features.7.conv.6.weight", "features.7.conv.7.weight", "features.7.conv.7.bias", "features.7.conv.7.running_mean", "features.7.conv.7.running_var", "features.8.conv.0.weight", "features.8.conv.1.weight", "features.8.conv.1.bias", "features.8.conv.1.running_mean", "features.8.conv.1.running_var", "features.8.conv.3.weight", "features.8.conv.4.weight", "features.8.conv.4.bias", "features.8.conv.4.running_mean", "features.8.conv.4.running_var", "features.8.conv.6.weight", "features.8.conv.7.weight", "features.8.conv.7.bias", "features.8.conv.7.running_mean", "features.8.conv.7.running_var", "features.9.conv.0.weight", "features.9.conv.1.weight", "features.9.conv.1.bias", "features.9.conv.1.running_mean", "features.9.conv.1.running_var", "features.9.conv.3.weight", "features.9.conv.4.weight", "features.9.conv.4.bias", "features.9.conv.4.running_mean", "features.9.conv.4.running_var", "features.9.conv.6.weight", "features.9.conv.7.weight", "features.9.conv.7.bias", "features.9.conv.7.running_mean", "features.9.conv.7.running_var", "features.10.conv.0.weight", "features.10.conv.1.weight", "features.10.conv.1.bias", "features.10.conv.1.running_mean", "features.10.conv.1.running_var", "features.10.conv.3.weight", "features.10.conv.4.weight", "features.10.conv.4.bias", "features.10.conv.4.running_mean", "features.10.conv.4.running_var", "features.10.conv.6.weight", "features.10.conv.7.weight", "features.10.conv.7.bias", "features.10.conv.7.running_mean", "features.10.conv.7.running_var", "features.11.conv.0.weight", "features.11.conv.1.weight", "features.11.conv.1.bias", "features.11.conv.1.running_mean", "features.11.conv.1.running_var", "features.11.conv.3.weight", "features.11.conv.4.weight", "features.11.conv.4.bias", "features.11.conv.4.running_mean", "features.11.conv.4.running_var", "features.11.conv.6.weight", "features.11.conv.7.weight", "features.11.conv.7.bias", "features.11.conv.7.running_mean", "features.11.conv.7.running_var", "features.12.conv.0.weight", "features.12.conv.1.weight", "features.12.conv.1.bias", "features.12.conv.1.running_mean", "features.12.conv.1.running_var", "features.12.conv.3.weight", "features.12.conv.4.weight", "features.12.conv.4.bias", "features.12.conv.4.running_mean", "features.12.conv.4.running_var", "features.12.conv.6.weight", "features.12.conv.7.weight", "features.12.conv.7.bias", "features.12.conv.7.running_mean", "features.12.conv.7.running_var", "features.13.conv.0.weight", "features.13.conv.1.weight", "features.13.conv.1.bias", "features.13.conv.1.running_mean", "features.13.conv.1.running_var", "features.13.conv.3.weight", "features.13.conv.4.weight", "features.13.conv.4.bias", "features.13.conv.4.running_mean", "features.13.conv.4.running_var", "features.13.conv.6.weight", "features.13.conv.7.weight", "features.13.conv.7.bias", "features.13.conv.7.running_mean", "features.13.conv.7.running_var", "features.14.conv.0.weight", "features.14.conv.1.weight", "features.14.conv.1.bias", "features.14.conv.1.running_mean", "features.14.conv.1.running_var", "features.14.conv.3.weight", "features.14.conv.4.weight", "features.14.conv.4.bias", "features.14.conv.4.running_mean", "features.14.conv.4.running_var", "features.14.conv.6.weight", "features.14.conv.7.weight", "features.14.conv.7.bias", "features.14.conv.7.running_mean", "features.14.conv.7.running_var", "features.15.conv.0.weight", "features.15.conv.1.weight", "features.15.conv.1.bias", "features.15.conv.1.running_mean", "features.15.conv.1.running_var", "features.15.conv.3.weight", "features.15.conv.4.weight", "features.15.conv.4.bias", "features.15.conv.4.running_mean", "features.15.conv.4.running_var", "features.15.conv.6.weight", "features.15.conv.7.weight", "features.15.conv.7.bias", "features.15.conv.7.running_mean", "features.15.conv.7.running_var", "features.16.conv.0.weight", "features.16.conv.1.weight", "features.16.conv.1.bias", "features.16.conv.1.running_mean", "features.16.conv.1.running_var", "features.16.conv.3.weight", "features.16.conv.4.weight", "features.16.conv.4.bias", "features.16.conv.4.running_mean", "features.16.conv.4.running_var", "features.16.conv.6.weight", "features.16.conv.7.weight", "features.16.conv.7.bias", "features.16.conv.7.running_mean", "features.16.conv.7.running_var", "features.17.conv.0.weight", "features.17.conv.1.weight", "features.17.conv.1.bias", "features.17.conv.1.running_mean", "features.17.conv.1.running_var", "features.17.conv.3.weight", "features.17.conv.4.weight", "features.17.conv.4.bias", "features.17.conv.4.running_mean", "features.17.conv.4.running_var", "features.17.conv.6.weight", "features.17.conv.7.weight", "features.17.conv.7.bias", "features.17.conv.7.running_mean", "features.17.conv.7.running_var", "features.18.0.weight", "features.18.1.weight", "features.18.1.bias", "features.18.1.running_mean", "features.18.1.running_var", "classifier.1.weight", "classifier.1.bias".

[limingzou-haha]

maybe this will help:
revise "model.load_state_dict(state_dict)" to "model.load_state_dict(state_dict,False)"

[nikitos4319]

Looks like there are weights for backbone only. It's just pretrained mnet2 and no trained model weights