Custom Dataset Training

[shixuan7]

Thank you for your excellent work and for sharing it with the community! I am very impressed by the pipeline you proposed. I was wondering, if I would like to train your pipeline on my own dataset for a segmentation task, which configuration files would I need to modify, and how should I adjust them? Any guidance would be greatly appreciated. Thank you!

[guangkaixu]

@shixuan7 Hi, for the training of the segmentation task, you can refer to this config. It regards the segmentation map as a 3-channel color map and directly encodes and decodes this colormap with VAE autoencoder. Another choice can be replacing the dpt decoder of this config with a segmentation head.