pytorch-semseg
pytorch-semseg copied to clipboard
How to use own data for training?
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.
Hi,
Is your problem solved? I want to know how to train our own dataset.
Thank you
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.
hi i would like to know how to train your own dataset,thank you
我已收到,谢谢