jiwoon-ahn / psa

Learning Pixel-level Semantic Affinity with Image-level Supervision for Weakly Supervised Semantic Segmentation, CVPR 2018
MIT License
380 stars 62 forks source link

Last fully connect layer #27

Open weixuansun opened 3 years ago

weixuansun commented 3 years ago

Hi, I notice that the last fc layer in resnet38_cls.py is defined as self.fc8 = nn.Conv2d(4096, 20, 1, bias=False), while I found another implementation define the last layer as a linear self.fc = nn.Linear(2048, num_classes), I wonder why do you use convolution instead, and is there any difference?