ALDM icon indicating copy to clipboard operation
ALDM copied to clipboard

Providing checkpoints for pretrained semantic segmentation models

Open samar-fathallah opened this issue 1 year ago • 1 comments

Hello there, In the paper it is mentioned that you trained the semantic segmentation models i.e. HRnet and SegFormer on cityscapes dataset augmented with diverse synthetic data generated by different methods. Are these pretrained models available anywhere for inference use (weights and checkpoints). I would like to adapt a similar evaluation approach for my task and it would be quite helpful to use those pretrained models. Thank you in advance.

samar-fathallah avatar Nov 26 '24 01:11 samar-fathallah

Hi @samar-fathallah , thanks for your interest!

Unfortunately, we are not able to release the segmentation related codes and models. But you could use mmsegmentation to train and test semantic segmentation models. We have released checkpoints for ALDM models, so you could generate new samples and train the segmenter with synthetic data.

YumengLi007 avatar Nov 26 '24 07:11 YumengLi007