4uiiurz1/pytorch-nested-unet

how to train on my own database?

manvirvirk opened this issue · 4 comments

can anyone help??

do you train dsb2018 success?

no, i want to train on retina database. i have small database containing retina images.

@manvirvirk @404leng
I add support for multi-class segmentation. Please pull the latest commit and try again. Let me know if you have any problems.

It may be dumb, but must my own dataset contains .jpg or .png files only? how about .nii or .nii.zip files? cuz I tried to train dataset with .nii.zip files, but it failed. Hope someone could help me. Appreciated!