Awesome-pytorch-list （ 精选 PyTorch 实用资源 ）

自然语言处理 (NLP)

• transformers: Huggingface 提供的最先进 NLP 模型库，支持 BERT、GPT 等模型。
• OpenNMT-py: 开源神经机器翻译工具。
• AllenNLP: 基于 PyTorch 的 NLP 研究工具。
• neuralcoref: 基于 spaCy 的核心指代消解工具。
• fairseq: Facebook 提供的序列建模工具。
• pytext: 自然语言建模框架。
• BERT-PyTorch: BERT 的 PyTorch 实现。

---

计算机视觉 (CV)

• torchvision: 提供预训练模型、图像数据集和数据增强工具。
• MMDetection: 强大的目标检测工具箱。
• detectron2: 下一代目标检测平台。
• albumentations: 快速图像增强库。
• kornia: 可微分的计算机视觉库。
• facenet-pytorch: 人脸检测和识别工具。
• MMSegmentation: 语义分割工具箱。
• pytorch3d: 3D 数据深度学习库。

---

语音处理

• speechbrain: 基于 PyTorch 的开源语音工具包。
• espnet: 端到端语音处理工具。
• TTS: 文本转语音工具。
• Tacotron2: 高效文本到语音生成工具。

---

深度学习工具

• pytorch-lightning: 提高 PyTorch 代码可读性和训练效率的框架。
• ignite: 高级训练工具库。
• fastai: 快速构建深度学习模型的工具。
• torchmetrics: 模型评估指标工具。

---

图神经网络

• PyG: 图深度学习扩展库。
• DGL: 深度图学习框架。

---

推荐教程

• PyTorch 官方教程: 官方提供的从入门到进阶的教程。
• pytorch-tutorial: 适合研究人员的 PyTorch 深度学习教程。
• Deep Learning with PyTorch: 《PyTorch 深度学习》书籍案例。
• Dive into Deep Learning (PyTorch): 《动手学深度学习》PyTorch 版。

---

常见论文实现

• transformers: Attention is All You Need 的实现。
• pix2pix: 图像到图像翻译的经典实现。
• CycleGAN: 无监督图像到图像翻译。
• DeepSpeech2: 语音识别模型 DeepSpeech2 的实现。
• faster-rcnn.pytorch: Faster R-CNN 的 PyTorch 实现。

---

实用工具

• tensorboard-pytorch: TensorBoard 的 PyTorch 插件。
• torchsummary: 模型结构和参数总结工具。
• torchstat: 模型分析工具。
• pytorch-OpCounter: 计算模型的 FLOPs。