search

MrGiovanni / UNetPlusPlus

[IEEE TMI] Official Implementation for UNet++
Other
2.26k stars 538 forks source link

Model shape error #61

Open Furqanf9T opened 3 years ago

Furqanf9T commented 3 years ago

Given transposed=1, weight of size [256, 128, 2, 2], expected input[1, 64, 128, 128] to have 256 channels, but got 64 channels instead ` network = Generic_UNetPlusPlus(num_input_channels, base_num_features, num_classes, len(net_num_pool_op_kernel_sizes), 2, 2, nn.Conv2d, nn.InstanceNorm2d, norm_op_kwargs, nn.Dropout2d, dropout_op_kwargs, net_nonlin, net_nonlin_kwargs, True, False, lambda x: x, InitWeights_He(1e-2), net_num_pool_op_kernel_sizes, net_conv_kernel_sizes, False, True, True)

a = torch.randn(1, 1, 256, 256)

b = network(a) ` image

HongliWu commented 3 years ago

Hi, friend! : ) i meet the same problem. Are you have solution of this issue?