deep_gcns
deep_gcns 是一个基于 TensorFlow 实现的开源项目,旨在解决图卷积网络(GCN)难以像卷积神经网络(CNN)那样构建深层架构的难题。在传统应用中,GCN 一旦层数加深,往往会出现梯度消失或过平滑现象,导致性能下降。deep_gcns 创新性地将 CNN 领域的成熟技术——包括残差连接、密集连接以及空洞卷积——迁移并适配到图神经网络结构中,成功实现了深层 GCN 的稳定训练与高效推理。
该项目不仅提供了完整的理论验证,还包含了针对 3D 点云语义分割任务(如 S3DIS 数据集)的训练、评估及可视化代码,并支持多种主流图卷积算子(如 MRGCN、EdgeConv、GraphSage 和 GIN)的对比实验。其核心亮点在于通过引入深度架构组件,显著提升了模型在复杂图结构数据上的特征提取能力。
deep_gcns 非常适合从事计算机视觉、几何深度学习的研究人员,以及需要处理 3D 点云数据的算法开发者使用。对于希望探索图神经网络深度化潜力或复现 ICCV 2019 口头报告成果的专业人士而言,这是一个极具参考价值的基准代码库。项目遵循 MIT 协议开源,文档详尽且环境配置便捷,有助于用户快速上手并进行二次开发。
使用场景
某自动驾驶团队正在利用激光雷达点云数据训练模型,以实现对城市街道中行人、车辆及路标的精准语义分割。
没有 deep_gcns 时
- 网络深度受限:传统图卷积网络(GCN)在堆叠超过几层后会出现严重的梯度消失或过平滑问题,导致无法构建深层架构来提取高阶特征。
- 局部细节丢失:浅层网络难以捕捉点云中长距离的空间依赖关系,导致对复杂场景(如被遮挡的物体)理解能力不足。
- 分割精度瓶颈:在 S3DIS 等大规模室内/室外数据集上,米级误差较多,细小物体(如交通标志杆)的识别率长期停滞不前。
- 调参试错成本高:工程师需花费大量时间手动调整邻域选择或尝试不同的基础算子,却难以突破性能天花板。
使用 deep_gcns 后
- 成功训练深层网络:借助残差连接、密集连接和膨胀卷积技术,团队轻松构建了数十层深的 GCN 模型,有效缓解了过平滑现象。
- 感受野显著扩大:通过引入类似 CNN 的膨胀机制,模型能聚合更广泛邻域的信息,显著提升了对全局上下文和长距离关系的建模能力。
- 分割精度大幅跃升:在相同数据条件下,平均交并比(mIoU)显著提升,尤其是对边缘模糊和被遮挡物体的分类更加精准。
- 研发效率提高:直接复用成熟的 DeepGCNs 架构模块,减少了重复造轮子的时间,让团队能专注于业务逻辑优化而非基础网络调试。
deep_gcns 通过将 CNN 的深度优势迁移至图神经网络,彻底打破了 GCN 难以加深的桎梏,让点云语义分割达到了前所未有的精度高度。
运行环境要求
- 未说明
未说明 (基于 TensorFlow 1.12.0,通常建议配备支持 CUDA 的 NVIDIA GPU)
未说明

快速开始
DeepGCNs:GCN 能否像 CNN 一样深?
在本工作中,我们提出了成功训练超深层 GCN 的新方法。我们借鉴了 CNN 中的概念,主要包括残差/密集连接和空洞卷积,并将其适配到 GCN 架构中。通过大量实验,我们展示了这些深层 GCN 框架的积极效果。
[项目] [论文] [幻灯片] [TensorFlow 代码] [PyTorch 代码]
概述
我们进行了大量实验,以展示不同组件(层数、滤波器数量、最近邻数、膨胀率等)对 DeepGCNs 的影响。此外,我们还针对不同类型的深层 GCN(MRGCN、EdgeConv、GraphSage 和 GIN)进行了消融研究。
如需更多信息和详情,请联系 Guohao Li 和 Matthias Müller。
需求
- TensorFlow 1.12.0
- h5py
- vtk(仅用于可视化)
- jupyter notebook(仅用于可视化)
Conda 环境
要设置包含所有必要依赖的 Conda 环境,请运行:
conda env create -f environment.yml
入门指南
您可以在 sem_seg 文件夹中找到关于我们代码如何用于三维点云语义分割的详细说明。目前,我们提供了以下内容:
- Conda 环境
- S3DIS 数据集 的设置
- 训练代码
- 评估代码
- 多个预训练模型
- 可视化代码
引用
如果您觉得我们的工作有所帮助,请引用我们的论文:
@InProceedings{li2019deepgcns,
title={DeepGCNs: Can GCNs Go as Deep as CNNs?},
author={Guohao Li and Matthias Müller and Ali Thabet and Bernard Ghanem},
booktitle={The IEEE International Conference on Computer Vision (ICCV)},
year={2019}
}
@misc{li2019deepgcns_journal,
title={DeepGCNs: Making GCNs Go as Deep as CNNs},
author={Guohao Li and Matthias Müller and Guocheng Qian and Itzel C. Delgadillo and Abdulellah Abualshour and Ali Thabet and Bernard Ghanem},
year={2019},
eprint={1910.06849},
archivePrefix={arXiv},
primaryClass={cs.CV}
}
许可证
MIT 许可证
致谢
本代码大量借鉴了 PointNet 和 EdgeConv 的实现。同时,我们也感谢 3d-semantic-segmentation 提供的可视化代码。
常见问题
相似工具推荐
openclaw
OpenClaw 是一款专为个人打造的本地化 AI 助手,旨在让你在自己的设备上拥有完全可控的智能伙伴。它打破了传统 AI 助手局限于特定网页或应用的束缚,能够直接接入你日常使用的各类通讯渠道,包括微信、WhatsApp、Telegram、Discord、iMessage 等数十种平台。无论你在哪个聊天软件中发送消息,OpenClaw 都能即时响应,甚至支持在 macOS、iOS 和 Android 设备上进行语音交互,并提供实时的画布渲染功能供你操控。 这款工具主要解决了用户对数据隐私、响应速度以及“始终在线”体验的需求。通过将 AI 部署在本地,用户无需依赖云端服务即可享受快速、私密的智能辅助,真正实现了“你的数据,你做主”。其独特的技术亮点在于强大的网关架构,将控制平面与核心助手分离,确保跨平台通信的流畅性与扩展性。 OpenClaw 非常适合希望构建个性化工作流的技术爱好者、开发者,以及注重隐私保护且不愿被单一生态绑定的普通用户。只要具备基础的终端操作能力(支持 macOS、Linux 及 Windows WSL2),即可通过简单的命令行引导完成部署。如果你渴望拥有一个懂你
stable-diffusion-webui
stable-diffusion-webui 是一个基于 Gradio 构建的网页版操作界面,旨在让用户能够轻松地在本地运行和使用强大的 Stable Diffusion 图像生成模型。它解决了原始模型依赖命令行、操作门槛高且功能分散的痛点,将复杂的 AI 绘图流程整合进一个直观易用的图形化平台。 无论是希望快速上手的普通创作者、需要精细控制画面细节的设计师,还是想要深入探索模型潜力的开发者与研究人员,都能从中获益。其核心亮点在于极高的功能丰富度:不仅支持文生图、图生图、局部重绘(Inpainting)和外绘(Outpainting)等基础模式,还独创了注意力机制调整、提示词矩阵、负向提示词以及“高清修复”等高级功能。此外,它内置了 GFPGAN 和 CodeFormer 等人脸修复工具,支持多种神经网络放大算法,并允许用户通过插件系统无限扩展能力。即使是显存有限的设备,stable-diffusion-webui 也提供了相应的优化选项,让高质量的 AI 艺术创作变得触手可及。
everything-claude-code
everything-claude-code 是一套专为 AI 编程助手(如 Claude Code、Codex、Cursor 等)打造的高性能优化系统。它不仅仅是一组配置文件,而是一个经过长期实战打磨的完整框架,旨在解决 AI 代理在实际开发中面临的效率低下、记忆丢失、安全隐患及缺乏持续学习能力等核心痛点。 通过引入技能模块化、直觉增强、记忆持久化机制以及内置的安全扫描功能,everything-claude-code 能显著提升 AI 在复杂任务中的表现,帮助开发者构建更稳定、更智能的生产级 AI 代理。其独特的“研究优先”开发理念和针对 Token 消耗的优化策略,使得模型响应更快、成本更低,同时有效防御潜在的攻击向量。 这套工具特别适合软件开发者、AI 研究人员以及希望深度定制 AI 工作流的技术团队使用。无论您是在构建大型代码库,还是需要 AI 协助进行安全审计与自动化测试,everything-claude-code 都能提供强大的底层支持。作为一个曾荣获 Anthropic 黑客大奖的开源项目,它融合了多语言支持与丰富的实战钩子(hooks),让 AI 真正成长为懂上
ComfyUI
ComfyUI 是一款功能强大且高度模块化的视觉 AI 引擎,专为设计和执行复杂的 Stable Diffusion 图像生成流程而打造。它摒弃了传统的代码编写模式,采用直观的节点式流程图界面,让用户通过连接不同的功能模块即可构建个性化的生成管线。 这一设计巧妙解决了高级 AI 绘图工作流配置复杂、灵活性不足的痛点。用户无需具备编程背景,也能自由组合模型、调整参数并实时预览效果,轻松实现从基础文生图到多步骤高清修复等各类复杂任务。ComfyUI 拥有极佳的兼容性,不仅支持 Windows、macOS 和 Linux 全平台,还广泛适配 NVIDIA、AMD、Intel 及苹果 Silicon 等多种硬件架构,并率先支持 SDXL、Flux、SD3 等前沿模型。 无论是希望深入探索算法潜力的研究人员和开发者,还是追求极致创作自由度的设计师与资深 AI 绘画爱好者,ComfyUI 都能提供强大的支持。其独特的模块化架构允许社区不断扩展新功能,使其成为当前最灵活、生态最丰富的开源扩散模型工具之一,帮助用户将创意高效转化为现实。
gemini-cli
gemini-cli 是一款由谷歌推出的开源 AI 命令行工具,它将强大的 Gemini 大模型能力直接集成到用户的终端环境中。对于习惯在命令行工作的开发者而言,它提供了一条从输入提示词到获取模型响应的最短路径,无需切换窗口即可享受智能辅助。 这款工具主要解决了开发过程中频繁上下文切换的痛点,让用户能在熟悉的终端界面内直接完成代码理解、生成、调试以及自动化运维任务。无论是查询大型代码库、根据草图生成应用,还是执行复杂的 Git 操作,gemini-cli 都能通过自然语言指令高效处理。 它特别适合广大软件工程师、DevOps 人员及技术研究人员使用。其核心亮点包括支持高达 100 万 token 的超长上下文窗口,具备出色的逻辑推理能力;内置 Google 搜索、文件操作及 Shell 命令执行等实用工具;更独特的是,它支持 MCP(模型上下文协议),允许用户灵活扩展自定义集成,连接如图像生成等外部能力。此外,个人谷歌账号即可享受免费的额度支持,且项目基于 Apache 2.0 协议完全开源,是提升终端工作效率的理想助手。
markitdown
MarkItDown 是一款由微软 AutoGen 团队打造的轻量级 Python 工具,专为将各类文件高效转换为 Markdown 格式而设计。它支持 PDF、Word、Excel、PPT、图片(含 OCR)、音频(含语音转录)、HTML 乃至 YouTube 链接等多种格式的解析,能够精准提取文档中的标题、列表、表格和链接等关键结构信息。 在人工智能应用日益普及的今天,大语言模型(LLM)虽擅长处理文本,却难以直接读取复杂的二进制办公文档。MarkItDown 恰好解决了这一痛点,它将非结构化或半结构化的文件转化为模型“原生理解”且 Token 效率极高的 Markdown 格式,成为连接本地文件与 AI 分析 pipeline 的理想桥梁。此外,它还提供了 MCP(模型上下文协议)服务器,可无缝集成到 Claude Desktop 等 LLM 应用中。 这款工具特别适合开发者、数据科学家及 AI 研究人员使用,尤其是那些需要构建文档检索增强生成(RAG)系统、进行批量文本分析或希望让 AI 助手直接“阅读”本地文件的用户。虽然生成的内容也具备一定可读性,但其核心优势在于为机器