how to train model with custom data?

charlesCXK / TorchSemiSeg

[CVPR 2021] CPS: Semi-Supervised Semantic Segmentation with Cross Pseudo Supervision

MIT License

504 stars 74 forks source link

Closed wuyunxiangwyx closed 2 years ago

wuyunxiangwyx commented 2 years ago

Thank you for your job! I want to know how to train cps with my custom dataset, i should change code where it is?

charlesCXK commented 2 years ago

Hi, just change the Dataset Class such as here, and change something related to your dataset in the config file.