Missing key(s) in state_dict error when testing the pretrained models about medicalnet HOT 5 OPEN

Alright. Could you kindly show me how to add necessary segmentation heads to the backbone?

you need to change the 'shortcut' parameter in Resnet class settings when you use different depth of resnet.
At the same time, you need to comment out the self.conv_seg variable when init the class.

I have been struggling with this myself, if I comment out the self.conv_seg then I can't use the pretrained resnet_50_epoch model right? Also, I have an issue with the resnet_50_epoch model where it has 2 classes specified but I need 1 class only. Any ideas how to approach this?

I haven't test it on resnet50, but on resnet 10, resnet 18 and resnet34, comment out the self.conv_seg worked well. Attention you need to change to 'shortcut'. I don't have any idea of the following question if you only need you class maybe this pretrained model do not fit your needs.

from medicalnet.

Hi, you can't get the segmentation result when using the segmentation test code in the situation you mentioned since "resnet_50_23dataset.pth" and "resnet_50.pth" are just backbones without segmentation head.

from medicalnet.

Alright. Could you kindly show me how to add necessary segmentation heads to the backbone?

from medicalnet.

Alright. Could you kindly show me how to add necessary segmentation heads to the backbone?

you need to change the 'shortcut' parameter in Resnet class settings when you use different depth of resnet.

At the same time, you need to comment out the self.conv_seg variable when init the class.

from medicalnet.

Alright. Could you kindly show me how to add necessary segmentation heads to the backbone?

you need to change the 'shortcut' parameter in Resnet class settings when you use different depth of resnet.

At the same time, you need to comment out the self.conv_seg variable when init the class.

I have been struggling with this myself, if I comment out the self.conv_seg then I can't use the pretrained resnet_50_epoch model right? Also, I have an issue with the resnet_50_epoch model where it has 2 classes specified but I need 1 class only. Any ideas how to approach this?

from medicalnet.