where do you implement concatenation

Hi,

From the image of the architecture in README.md, it seems that a concatenation was implemented. But from your unet.py, I couldn't find where do concatenation was implemented? I have trouble implementing the concatenation in U-net, Could you please help me understand it?

Great Thanks.

Hi, thank you for your interest in my code
Concatenation of feature maps is in forward function:

brain-segmentation-pytorch/unet.py

Line 54 in 8ef2e2d

dec4 = torch.cat((dec4, enc4), dim=1)

Let me know if you have any other questions.

Best regards,
Mat

Thanks!