PyTorch资源

timesler/facenet-pytorch

5.2k

1.0k

85

基于 PyTorch 的人脸识别库，提供预训练的 MTCNN 人脸检测和 InceptionResnet 识别模型。支持快速部署与微调，包含完整检测识别流程示例。

nvidiagameworks/kaolin

5.1k

627

33

基于 PyTorch 的 3D 深度学习加速库，提供模块化可微渲染、快速表示转换、数据加载、3D 检查点、可微相机与光照 API、结构化点云加速及 Jupyter 交互式可视化等 GPU 优化操作。

open-mmlab/mmaction2

5.1k

1.4k

318

OpenMMLab 的下一代视频理解工具箱与基准，基于 Python，提供丰富的模型库和简洁的 API。

tencentmusic/cube-studio

5.1k

883

44

开源云原生一站式 AI 平台，支持 MLOps 全流程、分布式训练、超参搜索、推理服务 VGPU 虚拟化、大模型微调与多机推理，兼容国产硬件及多种框架。

csailvision/semantic-segmentation-pytorch

5.1k

1.1k

81

基于 PyTorch 的 MIT ADE20K 数据集语义分割/场景解析实现。支持 PSPNet、UPerNet、HRNet 等 SOTA 模型，提供预训练模型与 Colab 演示。特色包括同步批归一化、多 GPU 动态输入尺度及高效训练。

autogptq/autogptq

5.1k

541

272

基于 GPTQ 算法的易用 LLM 量化包，提供用户友好的 API。支持 int4 量化，提升推理速度并降低显存占用。已集成至 Transformers 等框架。