How to set custom RGB-T datasets sets for training

huaaaliu / RGBX_Semantic_Segmentation

MIT License

316 stars 30 forks source link

How to set custom RGB-T datasets sets for training #14

Closed 1600337881 closed 1 year ago

1600337881 commented 2 years ago

Dear Author,I stored the data set as the readme.md, but mine is RGB-T instead of RGB-D.Num_classes,classes_name ,image_size and so on has been modified.But the result of train.py is strange as below. QQ图片20221116160600 QQ图片20221116160625

jamycheung commented 1 year ago

Hi @1600337881 ,

thanks for your interest. I would like to help if you can post how you modified the dataloader for your custom RGBT dataset. Besides, please check the MFNet dataset which we use in this work as RGBT dataset.