usr922 / wseg

[CVPR'22] Weakly Supervised Semantic Segmentation by Pixel-to-Prototype Contrast
126 stars 14 forks source link

pseudo mask IoU after rw #2

Closed BinghaoLu closed 2 years ago

BinghaoLu commented 2 years ago

Hi, thank you for your work on this paper and code, when I was implementing your code of random walk on the CAM, the generated pseudo mask IoU is less than 60% following the instruction, is there something that I missed?

usr922 commented 2 years ago

Hi @BinghaoLu , thanks for your attention. What about score curve of the CAM generated by step1? The pseudo labels are evaluated on train set and the segmentation model is retrained on train+aug set. You can follow the original repo (https://github.com/jiwoon-ahn/psa) of AffinityNet for more details :)