meetps/pytorch-semseg

How to use own data for training?

keishatsai opened this issue · 4 comments

keishatsai commented

Hi,

I am wondering that how to train with my own dataset and mask. I have searched issues, but it seems that there was not any update for the customise dataset.

Thanks.

GutlapalliNikhil commented

Hi,

Is your problem solved? I want to know how to train our own dataset.

Thank you

kubitz commented

You need to write your own data loader like in the loader module. I recommend going through the code with a known dataset and trying to understand what it does. If you find this too difficult, you can also copy the file/labelling structure from one of the supported datasets and simply change the number of classes and resolution.

  • 👍1
twsha commented

hi
i would like to know how to train your own dataset,thank you

hjhjb commented