pytorch_geometric

图形神经网络开发库

PyTorch Geometric 图神经网络机器学习深度学习数据处理 Github 开源项目

PyTorch Geometric是一个基于PyTorch的图形神经网络库，旨在简化结构化数据的建模与训练流程。支持小批量和大规模图的处理，并提供全面的GPU加速、数据管道处理以及常用基准数据集。这使得它成为机器学习研究者和初学者理想的选择。

Github

文档

论文

介绍相关项目

meshgpt-pytorch - 基于注意力机制的先进3D网格生成框架

3D建模GithubMeshGPT开源项目深度学习神经网络计算机图形学

MeshGPT-Pytorch是一个开源项目，专注于利用注意力机制实现3D网格生成。它基于PyTorch开发，支持可变长度面处理，并提供自动编码器和转换器模型。该项目计划引入文本条件控制功能，实现从文本到3D模型的转换。通过文本条件生成和分层转换器等高级特性，MeshGPT-Pytorch为3D内容创作和研究领域提供了先进的技术支持。

ogb - OGB开源项目的图机器学习数据集及标准评估工具

GithubOpen Graph Benchmark图机器学习开源项目数据加载器深度学习框架评估器

OGB项目集成了图机器学习所需的基准数据集、数据加载器和标准化评估工具，兼容PyTorch Geometric和DGL等主流深度学习框架。支持处理节点、链接和图级别的预测任务，数据集广泛涵盖科学、社交网络和异构知识图等领域，并适应从单GPU到多GPU的处理需求。OGB为研究人员提供了简化数据下载、标准化拆分和性能评估的便利工具。

pytorch-generative - PyTorch生成模型开发的全能助手

GithubPyTorch图像生成开源项目机器学习深度学习生成模型

pytorch-generative是一个强大的Python库，为PyTorch生成模型开发提供全方位支持。该库包含SOTA生成模型的参考实现、常用模块的抽象、实用的训练调试工具，以及TensorBoard集成功能。支持自回归模型、变分自编码器等多种算法，并提供简洁API和完善文档，有效提升生成模型的开发和复现效率。

geoopt - 黎曼流形优化的PyTorch扩展库

GithubPyTorchgeoopt开源项目流形深度学习黎曼优化

geoopt是一个专门用于黎曼流形优化的PyTorch扩展库。它提供多种黎曼流形、优化器和采样器,实现流形感知优化。该库支持自定义张量和参数,包括投影、平行传输等流形操作,以及RiemannianSGD、RiemannianAdam等优化器。geoopt适用于非欧几里得空间中的各类机器学习任务优化。

Open3D - 支持快速开发的3D数据处理开源库

3D数据处理C++ APIGPU加速GithubOpen3DPython API开源项目

支持快速开发3D数据处理应用的开源库，提供C++和Python接口。核心功能包括3D数据结构、3D数据处理算法、场景重建、表面对齐、3D可视化、基于物理的渲染（PBR）、3D机器学习支持（与PyTorch和TensorFlow兼容）、核心3D操作的GPU加速。适用于Ubuntu、macOS和Windows平台，支持源码编译和pip安装。

semantic-segmentation - 提供丰富数据集和易于定制的语义分割模型

GithubPyTorchSOTASemantic Segmentation开源项目数据集模型库

该项目提供易于使用和定制的SOTA语义分割模型，支持多种任务和数据集。适合高精度和定制应用场景，涵盖场景解析、人类解析、人脸解析等任务。特点包括多种主干网络和分割模型，支持PyTorch、ONNX、TFLite等框架的推理和导出。即将迎来重大更新，包括新的训练流程、预训练模型、教程和分布式训练支持。用户可通过详细文档和示例轻松使用并配置定制数据集，实现高效的语义分割。

pytorch_tabular - 表格数据深度学习的简易化解决方案

GithubPyTorchPyTorch Lightning开源项目标签数据模型定制深度学习

PyTorch Tabular是一个开源库，旨在简化表格数据的深度学习应用。该库具有低阻力易用性、易于定制和可扩展性，基于PyTorch和PyTorch Lightning构建。提供如TabNet、NODE和GATE等多种模型选择，适用于研究和实际应用。详细的文档和简单的安装流程使用户能够快速上手和自定义模型，提高机器学习任务的效率和性能。

pytorch-frame - 模块化深度学习框架用于异构表格数据

GithubPyTorch Frame开源项目模块化框架深度学习神经网络表格数据

PyTorch Frame是一个为异构表格数据设计的深度学习框架，支持数值、分类、时间、文本和图像等多种列类型。它采用模块化架构，实现了先进的深度表格模型，并可与大型语言模型集成。该框架提供了便捷的mini-batch加载器、基准数据集和自定义数据接口，简化了表格数据的深度学习研究过程，适用于各层次研究人员。框架内置多个预实现的深度表格模型，如Trompt、FTTransformer和TabNet等，并提供与XGBoost等GBDT模型的性能对比基准。PyTorch Frame无缝集成于PyTorch生态系统，便于与其他PyTorch库协同使用，为端到端的深度学习研究提供了便利。

segmentation_models.pytorch - 基于PyTorch的神经网络图像分割库

GithubPyTorch图像分割开源项目神经网络编码器预训练模型

segmentation_models.pytorch 是一个基于 PyTorch 的图像分割库,提供9种分割模型架构和124种编码器。该库 API 简洁,支持预训练权重,并包含常用评估指标和损失函数。它适用于研究和实际应用中的各种图像分割任务,是图像分割领域的实用工具。

pytorch-deep-learning - 深入PyTorch的深度学习实用教程

GithubPyTorch开源项目深度学习神经网络计算机视觉迁移学习

本课程涵盖从基础到高级的深度学习概念，通过实践教学与丰富的视频材料，讲解PyTorch操作和应用。包括神经网络分类、计算机视觉和数据集处理等主题，适合希望深化机器学习理解和应用的学习者。课程包括最新的PyTorch 2.0教程，确保内容的时效性和专业性。

相关项目

推荐项目

豆包MarsCode

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

AI写歌

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

有言AI

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

Kimi

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

阿里绘蛙

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

吐司

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

SubCat字幕猫

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

美间AI

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

AIWritePaper论文写作

AIWritePaper论文写作是一站式AI论文写作辅助工具，简化了选题、文献检索至论文撰写的整个过程。通过简单设定，平台可快速生成高质量论文大纲和全文，配合图表、参考文献等一应俱全，同时提供开题报告和答辩PPT等增值服务，保障数据安全，有效提升写作效率和论文质量。

使用协议隐私政策广告服务

投诉举报邮箱: service@vectorlightyear.com