search

lhoyer / MIC

[CVPR23] Official Implementation of MIC: Masked Image Consistency for Context-Enhanced Domain Adaptation
261 stars 40 forks source link

Training on Custom Dataset #48

Closed OSSome01 closed 1 year ago

OSSome01 commented 1 year ago

I want to train the model for semantic segmentation on a custom dataset. What steps do I need to take to do it? I see that I'll have to write some scripts for my dataset and put them in under "MIC/seg/mmseg/datasets." I have done this but when start training i get "KeyError: 'MyDataset is not in the dataset registry'"

Are there any more steps I have to do? Is there a detailed guide for using MIC on a Custom Dataset?

lhoyer commented 1 year ago

You need to import your custom dataset in the folders init.py to fix this error message.

There is no guide for MIC specifically, but you can check out the documentation of mmseg for more details.