最佳PyTorch工具集合:AI开发者的实用资源

monodepth2

monodepth2

Monodepth2深度估计

基于自监督学习的单目深度估计实现

tch-rs

tch-rs

tch-rslibtorch

Rust语言的PyTorch C++ API接口

Awesome-PyTorch-Chinese

Awesome-PyTorch-Chinese

PyTorch教程

PyTorch资源,教程、视频、实战项目和书籍推荐

pytorch-doc-zh

pytorch-doc-zh

PyTorch深度学习

PyTorch深度学习库中文文档与教程,支持GPU和CPU优化

d2l-pytorch

d2l-pytorch

Dive Into Deep LearningPyTorch

MXNet代码转换为PyTorch实现的指南

deep-person-reid

deep-person-reid

TorchreidPyTorch

深度学习人员重识别库,支持多GPU训练和跨数据集评估

mmocr

mmocr

OpenMMLabMMOCR

一个基于 PyTorch 和 mmdetection 的用于文本检测、文本识别以及相应的下游任务,包括关键信息提取的开源工具箱

dlwpt-code

dlwpt-code

Deep Learning with PyTorch深度学习

深入浅出PyTorch深度学习指南

DeepLearningProject

DeepLearningProject

Harvard UniversityPyTorch

全面教程涵盖数据集创建与深度学习

composer

composer

MosaicMLComposer

适用于大规模模型的高效深度学习训练框架

ssd.pytorch

ssd.pytorch

SSDPyTorch

PyTorch实现的高效SSD目标检测器,兼容多数据集与实时可视化

deep-learning-v2-pytorch

deep-learning-v2-pytorch

Deep LearningPyTorch

深度学习教程与项目实战指南

MONAI

MONAI

MONAIPyTorch

基于PyTorch的医疗影像深度学习开源平台

PyTorch-VAE

PyTorch-VAE

PyTorch VAE变分自编码器

PyTorch中多种变分自编码器的实现与训练示例

OpenNMT-py

OpenNMT-py

OpenNMT-pyEole

开源的神经机器翻译与大型语言模型框架

pyro

pyro

PyroPyTorch

灵活且可扩展的概率编程库,支持大规模数据处理与自定义推理

attention-is-all-you-need-pytorch

attention-is-all-you-need-pytorch

Transformer modelPyTorch

PyTorch版Transformer模型,采用自注意力机制

Pytorch-UNet

Pytorch-UNet

U-NetPyTorch

PyTorch实现的高效U-Net语义分割模型

YOLOX

YOLOX

YOLOXPyTorch

无锚目标检测算法YOLOX,设计简洁性能优越

pix2pix

pix2pix

pix2pixImage-to-Image Translation

利用条件对抗网络的图像到图像翻译实现