huawei-noah / Efficient-AI-Backbones

Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

关于代码中y的疑问

nhw649 opened this issue · comments

commented

你好,我想知道torch_vertex.py 146行,y代表什么,有什么作用呢?

if self.r > 1:
    y = F.avg_pool2d(x, kernel_size=self.r, stride=self.r)  # [B, out_dim, H/r, W/r]
    y = y.reshape(B, C, -1, 1).contiguous()  # [B, out_dim, H/r*W/r, 1]

y是为了减少节点个数,从而减少计算邻接矩阵时的计算量。

commented

y是为了减少节点个数,从而减少计算邻接矩阵时的计算量。

谢谢回答!

请问一下ViG-Ti在224x224下的FPS怎么样?相比于CNN、Transformer方法比较如何?

作者您好,您的论文非常有创意!但我想请问一下:
针对pyramid vig,如果r>1,下采样图会包含更大的感受野,而图更新来自特征图与下采样图(采样节点)的特征差值,这是否会带来除了减少计算外的某些好处?比如reduce_ratio元素全>1时,会不会比部分元素取1带来准确率的提升