UltraLight-VM-UNet
UltraLight-VM-UNet copied to clipboard
Prepare the data set
What format should your data set be formatted as input?
Hi! The data input in our project is the input of '.npy' format file. We have just updated the project interface with steps on how to preprocess with your own dataset, you can follow the steps to prepare your dataset. Hope this will help you.
Thank you very much for your help! What if I have multiple prospects, multiple object segmentation? How do you set up your own data?
Because I see that the data you give are single-target segmentation, there is only one prospect
The multi-objective segmentation task can be referred to the preparation and use of 'Synapse datasets' in VM-UNet. Also, replace 'vmunet.py' with 'UltraLight_VM-UNet.py' and change the output 'return torch.sigmoid(out0)' to 'return out0'.