MrGiovanni / UNetPlusPlus

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

How do I train U-Net++ on my custom 3D data? #94

Open Vishu9519 opened 7 months ago

Vishu9519 commented 7 months ago

Hi, I am having a hard time trying to modify the code so that I can train U-Net++ for my 3D data with 1 channel input and 1 channel output. For instance, the shape of input and output would be 16x256x256x1 (Depth,H,W,C). I do not need any backbone; I want to use the architecture of U-NET++.