search

YudeWang / semantic-segmentation-codebase

Codebase for semantic segmentation experiments
MIT License
56 stars 11 forks source link

how to generate pseudo label? #8

Open lions-heart opened 2 years ago

lions-heart commented 2 years ago

This is a great base work . I was learning how to use cam for segmentation training, and then found it. But I didn't find the function to generate pseudo label through cam.

YudeWang commented 2 years ago

Hi @lions-heart , This repo contains the code for retraining step. Please refer to other works such as https://github.com/YudeWang/SEAM for pseudo label generation.

lions-heart commented 2 years ago

Thank you for your reply. Is out_cam_pred the pseudo mask we want to generate? Cause i don`t find 'config.py' in SEAM.

YudeWang commented 2 years ago

That's right. Suggest to use the predictions of the random walk step, which is better. https://github.com/YudeWang/SEAM/blob/c55601649c5fa676836d3ec70ec044541b7d1d83/infer_aff.py#L60 More details please refer to README in SEAM repo.