search

mateuszbuda / brain-segmentation-pytorch

U-Net implementation in PyTorch for FLAIR abnormality segmentation in brain MRI
https://mateuszbuda.github.io/2017/12/01/brainseg.html
MIT License
722 stars 187 forks source link

how to train on my dataset #13

Open manvirvirk opened 4 years ago

mateuszbuda commented 4 years ago

Hi @manvirvirk If you want to run training on your dataset, you should be able to do this by implementing pytorch Dataset interface and replace BrainSegmentationDataset with it.

manvirvirk commented 4 years ago

i want to run training on simple dataset of retina images. can you please elaborate in detail???

manvirvirk commented 4 years ago

hi , I will be grateful if you can help me with little more details. How to do the above mentioned changes? Thanks

On Thu, Mar 19, 2020 at 8:41 PM mateuszbuda notifications@github.com wrote:

Hi @manvirvirk https://github.com/manvirvirk If you want to run training on your dataset, you should be able to do this by implementing pytorch Dataset interface and replace BrainSegmentationDataset https://github.com/mateuszbuda/brain-segmentation-pytorch/blob/8ef2e2d423b67b53ec8113fc71a9b968bb0f66e7/dataset.py#L12 with it.

— You are receiving this because you were mentioned. Reply to this email directly, view it on GitHub https://github.com/mateuszbuda/brain-segmentation-pytorch/issues/13#issuecomment-601234048, or unsubscribe https://github.com/notifications/unsubscribe-auth/ANAEAQFTTAUSC6HQQDP5XGTRIIYZHANCNFSM4LPKQXTQ .

mateuszbuda commented 4 years ago

Here is a tutorial on how to implement custom dataset in pytorch: https://pytorch.org/tutorials/beginner/data_loading_tutorial.html

manvirvirk commented 4 years ago

i want to run on my database which contains 20 images and respective masks. What should be the dataset directory?