tkuanlun350 / 3DUnet-Tensorflow-Brats18

3D Unet biomedical segmentation model powered by tensorpack with fast io speed
202 stars 68 forks source link

dataset structure #23

Open pervinh opened 5 years ago

pervinh commented 5 years ago

Can you tell me how dataset folder should look like? Mine looks like:

--\data ----\dataset -------\BRATS2018 -----------\training ---------------\HGG -------------------HGG147_Flair.nii -------------------HGG147_Label.nii -------------------HGG147_T1nii -------------------HGG147_T1c.nii -------------------HGG147_T2.nii ---------------\LGG -------------------LGG71_Flair.nii -------------------LGG71_Label.nii -------------------LGG71_T1nii -------------------LGG71_T1c.nii -------------------LGG71_T2.nii

But i get an error:

$ python preprocess.py                                                          
Processing HGG ...                                                               
0it [00:00, ?it/s]                                                               
Processing LGG ...                                                              
0it [00:00, ?it/s]
tkuanlun350 commented 5 years ago

For example: ls dataset/BRATS2018/training/HGG/Brats18_2013_10_1/ => Brats18_2013_10_1_flair.nii.gz Brats18_2013_10_1_seg.nii.gz Brats18_2013_10_1_t1ce.nii.gz Brats18_2013_10_1_t1.nii.gz Brats18_2013_10_1_t2.nii.gz

You can also modify dataloader.py to suit your need.