search

halbielee / EPS

Official PyTorch implementation of "Railroad is not a Train: Saliency as Pseudo-pixel Supervision for Weakly Supervised Semantic Segmentation", CVPR2021
89 stars 13 forks source link

Deeplab-v2 pretrain model #4

Closed allenwu97 closed 3 years ago

allenwu97 commented 3 years ago

I want to know which pretrain model you used when you train your deeplab-v2, imagenet or ms-coco? Thanks

halbielee commented 3 years ago

Hello, @allenwu97.

Sorry for the late reply.

There are two backbones for the DeepLab-V2; VGG-16 and ResNet101.

We used the official pre-trained model of DeepLab for VGG-16, which is trained on ImageNet, and that from the official code of OAA for fair comparisons. In the official code of OAA, there is no explanation of a source of the pre-trained model. But we assume that it is trained on MS-COCO.

A concurrent work with our paper, Anti-Adversarially Manipulated Attributions for Weakly and Semi-Supervised Semantic Segmentation (AdvCAM), provides the pre-trained model trained on ImageNet for ResNet101. Please see the official code of AdvCAM.

If you have any questions, please leave an issue :)

Thanks.

allenwu97 commented 3 years ago

So for your ResNet101, you use the pre-trained model from the official code of OAA?

halbielee commented 3 years ago

@allenwu97 Yes!

allenwu97 commented 3 years ago

Thanks for your patient reply.