search

jbeomlee93 / RIB

Reducing Information Bottleneck for Weakly Supervised Semantic Segmentation (NeurIPS 2021)
87 stars 4 forks source link

Training coco part code is missing. #1

Open TyroneLi opened 3 years ago

TyroneLi commented 3 years ago

In folder step, train_cam_coco.py is missing. Can you upload it? Thans

TyroneLi commented 3 years ago

By the way, your code for refining the initial cam see using CRF is not yet released.

TyroneLi commented 3 years ago

Hello~,anybody~

jbeomlee93 commented 3 years ago

Sorry for the late reply.

  1. I have updated "step/train_cam_coco.py", but it has not been newly tested.

You can run RIB process with pre-trained classifier provided by https://drive.google.com/file/d/1SDKjPzzuXR4PX_H1l3Kw5NY8d20FmG4X/view.

  1. You can refine the initial cam using CRF similar to below.

https://github.com/jbeomlee93/RIB/blob/9c43b3b170f49d46dfb54213f7da57160ab20715/step/cam_to_ir_label.py#L21-L35