pytorch-semseg icon indicating copy to clipboard operation
pytorch-semseg copied to clipboard

unet upsample-concatenate problem

Open Zakiyi opened this issue 6 years ago • 0 comments

Hello, thx for your sharing, I am new to segmentation with deep learning. I have checked the unet paper, in the decode part of unet, the upsampled feature maps seems reduced channel and then concatenated with corresponding feature from encoder part. hence, the channel doubled. however, in your code, the channel of upsampled feature (bilinear) directly concatenated with the encoder part feature, and then passed through unetConv2(two layers convolutional layers), so i wonder isn't wrong here? Supposing upsampled feature channel is 1024, the counterpart feature channel is 512, then the channel of concatenated feature should be 1536, thus it's conflicted with filters[4]=1024 "self.up_concat4 = unetUp(filters[4], filters[3])" filters[4]=1024 I am puzzled :(

Zakiyi avatar Jan 10 '19 14:01 Zakiyi