UltraLight-VM-UNet icon indicating copy to clipboard operation
UltraLight-VM-UNet copied to clipboard

Prepare the data set

Open liubaoning111 opened this issue 10 months ago • 4 comments

What format should your data set be formatted as input?

liubaoning111 avatar Apr 04 '24 10:04 liubaoning111

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.

wurenkai avatar Apr 04 '24 12:04 wurenkai

Thank you very much for your help! What if I have multiple prospects, multiple object segmentation? How do you set up your own data?

liubaoning111 avatar Apr 04 '24 13:04 liubaoning111

Because I see that the data you give are single-target segmentation, there is only one prospect

liubaoning111 avatar Apr 04 '24 13:04 liubaoning111

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'.

wurenkai avatar Apr 04 '24 15:04 wurenkai