brain-segmentation-pytorch icon indicating copy to clipboard operation
brain-segmentation-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon mateuszbuda

how to train on my dataset

Open manvirvirk opened this issue 4 years ago • 5 comments

manvirvirk avatar Mar 19 '20 14:03 manvirvirk

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.

mateuszbuda avatar Mar 19 '20 15:03 mateuszbuda

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

manvirvirk avatar Mar 19 '20 15:03 manvirvirk

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 [email protected] 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 .

manvirvirk avatar Mar 19 '20 19:03 manvirvirk

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

mateuszbuda avatar Mar 21 '20 12:03 mateuszbuda

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

manvirvirk avatar Apr 07 '20 13:04 manvirvirk