gt_classes in VOC2012.py

showlab / DatasetDM

[NeurIPS2023] DatasetDM:Synthesizing Data with Perception Annotations Using Diffusion Models

https://weijiawu.github.io/DatasetDM_page/

290 stars 12 forks source link

gt_classes in VOC2012.py #17

Closed SchrodingerLLX closed 6 months ago

SchrodingerLLX commented 8 months ago

你好，有个问题想请教一下，在VOC2012.py中gt_classes为什么是1而不是select_class？

instances = {} mapper_classes = [1] instances["gt_classes"] = torch.tensor(mapper_classes, dtype=torch.int64) masks = [] masks.append(mask == 1)

https://github.com/showlab/DatasetDM/blob/7545a68f35d8aad37f7963c326ae7c759a9a89d2/dataset/VOC2012.py#L233C2-L234C82 #

Z1zs commented 5 months ago

I have the same question, and I'm even more confused cause the class prediction of instance segmentation task is all "bicycle"(class 1). Have you figure it out?