用 PyTorch 复现了注意力机制、MLP、重参数化和卷积等常见模块,方便对照论文理解原理并直接嵌入视觉任务模型。
基于 PyTorch 的图神经网络库,提供图卷积等 GNN 模型的构建与训练组件,省去自己实现图数据结构和消息传递的麻烦。
基于 PyTorch 的语义分割模型库,提供 500+ 预训练卷积与 Transformer 骨干网络,开箱即用,省去自己搭建和训练分割模型的麻烦。
Facebook AI 开源的端到端语音识别训练工具包,用 C++ 实现,可基于 wav2letter 架构训练声学模型,省去传统 ASR 繁琐的特征工程流程。
用深度卷积网络自动提升照片画质,提供预训练模型和训练代码,可把手机照片修出单反质感。
Conformer 的 PyTorch 非官方实现,把卷积和 Transformer 结合起来做语音识别,适合直接拿来训练或复现 ASR 模型。
PointCNN 提出 X-变换卷积算子,直接在无序点云上做特征学习,可用于三维点云分类和分割,服务于自动驾驶、机器人等场景。
用稀疏 4D 卷积在 3D 激光雷达点云中做移动物体分割,能区分场景里的动与静,适合自动驾驶感知研究。
U-Time 是一个用全卷积网络做时间序列分割的模型,最初用于睡眠分期分析,也可迁移到其他时序分割任务。
MIT-Princeton 开源的 Amazon Picking Challenge 2016 视觉工具箱,用 RGB-D 卷积网络做物体分割和 6D 位姿估计,帮机器人完成抓取任务。