1 Vision GNN:把一张图片建模为一个图 (来自中国科学院大学,华为诺亚方舟实验室,北大) 1.1 Vision GNN 原理分析 1.1.1 背景和动机 1.1.2 图片的图表示 (Graph Representation) 1.1.3 使用 GNN 作为模型骨干 1.1.4 ViG Block 1.1.5 实验结果 1.1.6 可视化结果 1.2 Vision GNN PyTorch 伪代码
论文名称:Vision GNN: An Image is Worth Graph of Nodes
上式5中的 里面的 和 分别对应伪代码中的 self.fc1 和 self.fc2,二者都是由 nn.Conv2d(kernel=1) 进行实现,并在后面加上 nn.BatchNorm2d。 GCN 部分使用了 GitHub - lightaime/deep_gcns_torch: Pytorch Repo for DeepGCNs (ICCV'2019 Oral, TPAMI'2021), DeeperGCN (arXiv'2020) and GNN1000(ICML'2021): https://www.deepgcns.org 中的 DynConv2d(in_channels, hidden_channels, k, dilation, act=None),激活函数 使用的是 nn.GeLU()。
您需要 登录 才可以下载或查看,没有账号?立即注册
使用道具 举报
本版积分规则 发表回复 回帖并转播 回帖后跳转到最后一页
小黑屋|手机版|Unity开发者联盟 ( 粤ICP备20003399号 )
GMT+8, 2025-5-2 12:31 , Processed in 0.309192 second(s), 26 queries .
Powered by Discuz! X3.5 Licensed
© 2001-2025 Discuz! Team.