#353 · 主分类: 计算机视觉

deep_gcns_torch

3d-point-clouds bioinformatics cheminformatics computer-vision data-mining deep-gcns deep-learning geometric-deep-learning graph-convolutional-networks graph-neural-networks pytorch science-research social-network

DeepGCNs(ICCV'2019口头报告,TPAMI'2021)、DeeperGCN(arXiv'2020)和GNN1000(ICML'2021)的Pytorch代码库:https://www.deepgcns.org

项目最后更新:07/31/22

GitHub Stars

1.2K

Forks数量

153

贡献者数量

9

许可证

MIT

收录理由

这是DeepGCNs系列研究的PyTorch官方参考实现,该系列把残差连接、稠密连接、空洞卷积等CNN中的深度训练技巧迁移到图卷积网络上,并系统验证了这些设计对深层GCN训练的有效性。代码仓库为三维点云处理场景提供了可直接运行的完整流程:涵盖S3DIS与PartNet上的语义分割、ModelNet40上的分类任务,每个任务目录都附带数据、配置与预训练权重,开箱即用。此外还收录了DeeperGCN和GNN1000的后续变体,方便在OGB图数据集上对比不同深度与显存高效的图神经网络方案。核心图卷积层封装在模块化的gcn_lib中,同时支持稠密与稀疏两种后端,便于将底层算子抽取出来集成到自己的模型里,而不只是复现论文结果。

相关文章

暂无针对该项目的文章。

如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。

同类热门项目