How to train custom dataset?

NVIDIA / semantic-segmentation

Nvidia Semantic Segmentation monorepo

BSD 3-Clause "New" or "Revised" License

1.76k stars 388 forks source link

Closed yes89929 closed 2 years ago

yes89929 commented 2 years ago

Hi. Thanks you for shairing good code. But, I`m not familiar with deeplearning and pytorch.

I want to train your model using my custom datasets. And, My custom dataset has 4 classes with void class(0).

May I have some hint tor training my custom datasets?

ajtao commented 2 years ago

I'm sorry, it's like asking for a data science class. Why don't you try to learn from pytorch examples.