the format of the label image

[wwjwy]

Hi Luo,
Thank you very much for your codebase. I modified your code so that it can be used for weakly supervised semantic segmentation of RGB images. I encountered some problems when preparing the dataset. My dataset contains two categories, background and spalling. The mask I made is shown in the following image, black scribble(background) and white scribble(spalling). I want to know is my way of labeling correct?

[Luoxd1996]

Hi,
For the pCE loss, the background is labeled as 0 and the foreground is labeled as 1, the others is x which can be seen as ignore_index=x.
Best,
Xiangde.

[wwjwy]

Hi,
For the pCE loss, the background is labeled as 0 and the foreground is labeled as 1, the others is x which can be seen as ignore_index=x.
Best,
Xiangde.

In other words, for the mask, I only need to label a few scribbles or points, and the other parts in the image remain the same, right?

[Luoxd1996]

No, the mask just consists of [0,1,x], x means the unlabeled region. You'd better read some weakly-supervised papers.
Best,
Xiangde.

[wwjwy]

No, the mask just consists of [0,1,x], x means the unlabeled region. You'd better read some weakly-supervised papers.
Best,
Xiangde.

I got it, thank you!