search

MendelXu / SAN

Open-vocabulary Semantic Segmentation
https://mendelxu.github.io/SAN/
MIT License
307 stars 28 forks source link

about caption #43

Closed HarryMark927 closed 9 months ago

HarryMark927 commented 9 months ago

Where is the code that makes each image category name correspond to the labeling mask during training

MendelXu commented 9 months ago

Each mask has its related class id (each one is mapped to a category name). https://github.com/MendelXu/SAN/blob/81a9a2bd79d433292d46cfa0597caea5005e0116/san/data/dataset_mappers/mask_former_semantic_dataset_mapper.py#L176