How to extract augented segmentation mask for retrieving class identity

[Muhammad-Ibraheem-Siddiqui]

Hello,

Thanks for the great repo. I wanted to evaluate BAM using the pascal validation dataset. Now in the get_item function, I am not able to retrieve the class identity using the original label in the Pascal VOC dataset. I am accessing the label from the folder "segmentatioclass" but the mask present here does not contain class identity. Your code refers to a folder named as SegmentationClassAug" for retrieving the labels and subsequently class identity. This folder is not present in the Pascal VOC dataset. So, can you explain that how to get this augmented segmentation mask as a label for retrieval of class identity. I have also attached the relevant code snippet. Thanks.