datasets/prepare_ade20k_full_sem_seg.py mask_former/data/datasets/register_ade20k_full.py