Logo
Project Icon

equiformer-pytorch

SE(3)/E(3)等变注意力网络的高效PyTorch实现

EquiformerSE3 TransformersGATv2AI深度学习Github开源项目

Equiformer-pytorch是一个基于PyTorch的SE(3)/E(3)等变注意力网络实现。该项目采用MLP注意力机制和非线性消息传递，实现了最先进的性能。它支持可逆网络以提高内存效率，并集成了最新的球谐函数稀疏化技术，大幅提升计算效率。Equiformer-pytorch还提供边缘和邻接矩阵支持，适用于蛋白质折叠等各种3D原子图任务。

访问官网访问官网GithubGithub论文论文
介绍相关项目
ai-edge-torch - PyTorch模型转TensorFlow Lite的开源解决方案
AI Edge TorchGithubPyTorchTensorFlow Lite开源项目模型转换移动设备部署
ai-edge-torch是一个开源Python库，用于将PyTorch模型转换为TensorFlow Lite格式。它支持在Android、iOS和IoT设备上本地运行模型，提供广泛的CPU支持和初步的GPU、NPU支持。该项目还包含生成式API，用于优化大型语言模型在设备端的性能。ai-edge-torch与PyTorch紧密集成，为边缘AI开发提供了实用的工具。
3DMPPE_ROOTNET_RELEASE - 单张RGB图像的相机距离感知的3D多人人体姿态估计实现
3D姿态估计GithubPyTorchRGB图像RootNet多人体姿态估计开源项目
此项目基于PyTorch实现了3D多人人体姿态估计，兼容多种公开的2D和3D数据集，如Human3.6M、MPII、MS COCO、MuCo-3DHP、MuPoTS-3D和3DPW。其特点包括代码简洁灵活、直观的人体姿态可视化，并支持不同单位系统的适配。项目还提供详细的训练和测试指南，旨在帮助用户在GPU环境下高效运行姿态估计算法。
seemore - PyTorch实现的开源视觉语言模型项目
AIGithubPytorchVision Language Model图像处理开源项目机器学习
seemore是一个基于PyTorch的开源视觉语言模型(VLM)项目。它包括图像编码器、视觉-语言投影器和解码器三个核心组件，参考了CLIP和LLaVA等前沿VLM架构。项目提供完整代码实现和详细教程，有助于开发者理解VLM原理。seemore在Databricks平台开发，支持GPU加速，并集成MLFlow用于实验管理。
Transformer-TTS - 神经语音合成系统
GithubPyTorchTacotronTransformer-TTS开源项目神经网络语音合成
Transformer-TTS，一个基于Pytorch的高效神经语音合成系统。它使用Transformer网络，且训练速度是传统seq2seq模型的3到4倍。不仅提供预训练模型，其合成语音质量经实验证明优异。同时，项目支持自定义学习模型及策略，包括Noam式预热衰减学习率及关键的梯度裁剪等，是语音合成研究的理想选择。
gtsfm - 高性能并行结构运动恢复管线GTSfM
3D重建GTSAMGTSfMGithub并行计算开源项目结构运动恢复
GTSfM是一个基于GTSAM的开源结构运动恢复(SfM)管线，专为并行计算设计。它利用Dask实现分布式处理，集成了SuperPoint和SuperGlue等先进算法。GTSfM提供Python接口，无需编译即可使用。该项目支持多种场景重建任务，可与Nerfstudio等工具集成，为计算机视觉领域提供了灵活高效的解决方案。
iTransformer - 先进的时间序列预测模型，打造SOTA性能
GithubiTransformer人工智能开源项目时间序列预测注意力网络深度学习
iTransformer是一种基于注意力机制的时间序列预测模型，由清华大学和蚂蚁集团研究人员开发。该模型采用倒置Transformer结构，支持多变量和多步长预测。iTransformer引入了可逆实例归一化等技术，旨在提高预测准确性和处理长序列数据的能力。这个开源项目为时间序列分析提供了新的研究方向。项目提供Python实现，支持使用PyTorch框架。用户可通过pip安装并轻松集成到现有的时间序列分析工作流程中。该项目还包括实验性功能，如二维注意力和傅里叶变换增强版本，为研究人员提供了探索和改进的空间。
facenet-pytorch - Pytorch下基于预训练模型和快速MTCNN的人脸识别技术
Face RecognitionGithubInception Resnet V1MTCNNPytorchVGGFace2开源项目
facenet-pytorch是一个开源项目，利用Pytorch实现的Inception Resnet (V1)模型，已在VGGFace2和CASIA-Webface上预训练。项目还包括高效的MTCNN实现，用于人脸检测。其特点包括快速人脸检测、生成面部识别嵌入、视频流中的人脸跟踪以及从TensorFlow到Pytorch的参数转换。开发者可以通过简单的安装和使用指引快速集成和应用这些高精度、高性能的预训练模型和工具。
TransMorph_Transformer_for_Medical_Image_Registration - 基于Transformer的无监督医学图像配准方法
GithubPyTorchTransMorphTransformer医学影像配准开源项目深度学习
TransMorph是一个利用Transformer架构进行无监督医学图像配准的开源项目，结合了Vision Transformer和Swin Transformer技术。提供多个模型变体和多种损失函数，支持单模态和多模态配准，公开了训练脚本和预训练模型，并在MICCAI 2021 L2R挑战中表现出色。
make-a-video-pytorch - 基于 PyTorch 的最新文本到视频生成器
3D卷积GithubMake-A-VideoMeta AIPytorch开源项目时序注意力
此项目实现了 Meta AI 的 Make-A-Video 在 PyTorch 下的版本，利用伪 3D 卷积和时序注意力技术，显著增强了视频的时序一致性。支持图像和视频帧的处理，并且可轻松适用于 DALL-E2 和 Imagen 等模型。项目提供了完整的安装和使用指南，并支持空间和时间一致性的 Unet 模型。
flash-attention - 高效注意力机制加速深度学习模型训练
CUDAFlashAttentionGPU加速GithubPyTorch开源项目注意力机制
FlashAttention是一种高效的注意力机制实现,通过IO感知算法和内存优化提升计算速度并降低内存消耗。它支持NVIDIA和AMD GPU,适用于多种深度学习框架。最新的FlashAttention-3版本针对H100 GPU进行了优化。该项目提供Python接口,可集成到现有模型中,有助于加速大规模深度学习模型的训练过程。
相关项目
项目侧边栏1项目侧边栏2
推荐项目
Project Cover

豆包MarsCode

豆包 MarsCode 是一款革命性的编程助手，通过AI技术提供代码补全、单测生成、代码解释和智能问答等功能，支持100+编程语言，与主流编辑器无缝集成，显著提升开发效率和代码质量。

Project Cover

AI写歌

Suno AI是一个革命性的AI音乐创作平台，能在短短30秒内帮助用户创作出一首完整的歌曲。无论是寻找创作灵感还是需要快速制作音乐，Suno AI都是音乐爱好者和专业人士的理想选择。

Project Cover

有言AI

有言平台提供一站式AIGC视频创作解决方案，通过智能技术简化视频制作流程。无论是企业宣传还是个人分享，有言都能帮助用户快速、轻松地制作出专业级别的视频内容。

Project Cover

Kimi

Kimi AI助手提供多语言对话支持，能够阅读和理解用户上传的文件内容，解析网页信息，并结合搜索结果为用户提供详尽的答案。无论是日常咨询还是专业问题，Kimi都能以友好、专业的方式提供帮助。

Project Cover

阿里绘蛙

绘蛙是阿里巴巴集团推出的革命性AI电商营销平台。利用尖端人工智能技术，为商家提供一键生成商品图和营销文案的服务，显著提升内容创作效率和营销效果。适用于淘宝、天猫等电商平台，让商品第一时间被种草。

Project Cover

吐司

探索Tensor.Art平台的独特AI模型，免费访问各种图像生成与AI训练工具，从Stable Diffusion等基础模型开始，轻松实现创新图像生成。体验前沿的AI技术，推动个人和企业的创新发展。

Project Cover

SubCat字幕猫

SubCat字幕猫APP是一款创新的视频播放器，它将改变您观看视频的方式！SubCat结合了先进的人工智能技术，为您提供即时视频字幕翻译，无论是本地视频还是网络流媒体，让您轻松享受各种语言的内容。

Project Cover

美间AI

美间AI创意设计平台，利用前沿AI技术，为设计师和营销人员提供一站式设计解决方案。从智能海报到3D效果图，再到文案生成，美间让创意设计更简单、更高效。

Project Cover

稿定AI

稿定设计 是一个多功能的在线设计和创意平台，提供广泛的设计工具和资源，以满足不同用户的需求。从专业的图形设计师到普通用户，无论是进行图片处理、智能抠图、H5页面制作还是视频剪辑，稿定设计都能提供简单、高效的解决方案。该平台以其用户友好的界面和强大的功能集合，帮助用户轻松实现创意设计。

使用协议隐私政策广告服务
投诉举报邮箱: service@vectorlightyear.com
@2024 懂AI·鲁ICP备2024100362号-6·鲁公网安备37021002001498号