fudan-zvg / SeaFormer

[ICLR 2023] SeaFormer: Squeeze-enhanced Axial Transformer for Mobile Semantic Segmentation
290 stars 21 forks source link

关于代码的两个疑问 #13

Closed Ifjmww closed 1 year ago

Ifjmww commented 1 year ago

作者您好,您和您团队的工作非常出色,我对SeaFormer模型很感兴趣,在阅读完代码后有以下两个疑问: 1.https://github.com/fudan-zvg/SeaFormer/blob/cd1d0092503215349ae4670aa80b03dee4d9a385/seaformer-seg/mmseg/models/backbones/seaformer.py#L456-L463
为什么是在i==1时才存入输出结果,按照论文图示,应该是i==0是存入输出结果。

2.https://github.com/fudan-zvg/SeaFormer/blob/cd1d0092503215349ae4670aa80b03dee4d9a385/seaformer-seg/mmseg/models/decode_heads/light_head.py#L81 最后使用的cls_seg函数并没有进行BN操作,这与论文图中所示不符。

幻灯片1

以上两个问题您能否解答一下,不胜感激!

wwqq commented 1 year ago

@Ifjmww 您好 1.Shared STEM包含smb1和smb2,smb1四倍下采样,smb2二倍下采样。1/8下采样后的MV2↓2和MV2两个模块组成smb3。所以应该是i=1存入结果。 2.感谢纠正!这里只有conv。

Ifjmww commented 1 year ago

感谢您的回复!