kazuto1011 / deeplab-pytorch

PyTorch re-implementation of DeepLab v2 on COCO-Stuff / PASCAL VOC datasets

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

The training result is poor after using the weak supervision data set

woqiaow opened this issue · comments

Hello, if I want to use weakly supervised data to replace the ground truth value to train the segmented network in the experiment, do I need to modify the code after replacing the dataset?
Because I tried to run this code with the pseudo mask image (10582) obtained by weak supervision, but the result was only 0.03, which actually only recognized the background. Do I need to modify the code?
image

Have you checked if your GT files have been loaded successfully?
I think you may also need to debug the class-wise progress for training not only the overall loss.

Is the GT file you mentioned the ground truth image? Yes, successfully loaded. And I tried to train with train (1464) to produce results. The result was 61%, lower than expected. With train_aug (10582), it could only identify the background. The scores of other categories were 0.
Do you have any other suggestions besides debugging?
Thanks for your help!

From your results, I guess your model only produces 0 for any images.
Sorry for the duplicated question. Are your loaded weakly-supervised labels decoded numbers from 0 to 20 successfully? Does it suit your task that the default setting only treats 255 as the label to be ignored in the loss computation? Since I never assumed the weakly-supervised setting to maintain this repository, I can only suggest general things.

OK, I see. I will go to debug it. Thank you!