FCN-semantic-segmentation icon indicating copy to clipboard operation
FCN-semantic-segmentation copied to clipboard

ResNet50

Open buaaswf opened this issue 6 years ago • 2 comments

Hi@Kaixhin, I am trying to improve the resnet34 to ResNet50, but I met the reshape problems File "/data2/FCN-semantic-segmentation_new/model.py", line 92, in forward x = self.relu(self.bn8(self.conv8(x + x2))) RuntimeError: The size of tensor a (256) must match the size of tensor b (64) at non-single

Cloud you please help me?

buaaswf avatar Apr 01 '18 11:04 buaaswf

The error message indicates that you haven't matched up the downsampled versions of the input with the upsampled versions. If you are using a different network as the backbone for segmentation, you will need to match these up properly. An easy way to do this is to put print('conv_y', x.size()) statements in the forward pass of your new resnet, and then see at which points these are.

Kaixhin avatar Apr 01 '18 13:04 Kaixhin

@buaaswf 你好,这个代码你跑通了吗?

HPL123 avatar Jun 01 '18 13:06 HPL123