https://github.com/xiaomingx/awesome-pytorch-list
A comprehensive list of pytorch related content on github,such as different models,implementations,helper libraries,tutorials etc.
https://github.com/xiaomingx/awesome-pytorch-list
List: awesome-pytorch-list
Last synced: 3 months ago
JSON representation
A comprehensive list of pytorch related content on github,such as different models,implementations,helper libraries,tutorials etc.
- Host: GitHub
- URL: https://github.com/xiaomingx/awesome-pytorch-list
- Owner: XiaomingX
- License: apache-2.0
- Created: 2024-11-24T16:38:24.000Z (6 months ago)
- Default Branch: main
- Last Pushed: 2024-11-24T16:40:25.000Z (6 months ago)
- Last Synced: 2024-11-24T17:27:49.279Z (6 months ago)
- Size: 0 Bytes
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
- ultimate-awesome - awesome-pytorch-list - A comprehensive list of pytorch related content on GitHub,such as different models,implementations,helper libraries,tutorials etc. (Programming Language Lists / Python Lists)
README
# Awesome-pytorch-list （ 精选 PyTorch 实用资源 ）
## 自然语言处理 (NLP)
- [transformers](https://github.com/huggingface/transformers): Huggingface 提供的最先进 NLP 模型库，支持 BERT、GPT 等模型。
- [OpenNMT-py](https://github.com/OpenNMT/OpenNMT-py): 开源神经机器翻译工具。
- [AllenNLP](https://github.com/allenai/allennlp): 基于 PyTorch 的 NLP 研究工具。
- [neuralcoref](https://github.com/huggingface/neuralcoref): 基于 spaCy 的核心指代消解工具。
- [fairseq](https://github.com/facebookresearch/fairseq): Facebook 提供的序列建模工具。
- [pytext](https://github.com/facebookresearch/pytext): 自然语言建模框架。
- [BERT-PyTorch](https://github.com/codertimo/BERT-pytorch): BERT 的 PyTorch 实现。
---
## 计算机视觉 (CV)
- [torchvision](https://github.com/pytorch/vision): 提供预训练模型、图像数据集和数据增强工具。
- [MMDetection](https://github.com/open-mmlab/mmdetection): 强大的目标检测工具箱。
- [detectron2](https://github.com/facebookresearch/detectron2): 下一代目标检测平台。
- [albumentations](https://github.com/albumentations-team/albumentations): 快速图像增强库。
- [kornia](https://github.com/kornia/kornia): 可微分的计算机视觉库。
- [facenet-pytorch](https://github.com/timesler/facenet-pytorch): 人脸检测和识别工具。
- [MMSegmentation](https://github.com/open-mmlab/mmsegmentation): 语义分割工具箱。
- [pytorch3d](https://github.com/facebookresearch/pytorch3d): 3D 数据深度学习库。
---
## 语音处理
- [speechbrain](https://github.com/speechbrain/speechbrain): 基于 PyTorch 的开源语音工具包。
- [espnet](https://github.com/espnet/espnet): 端到端语音处理工具。
- [TTS](https://github.com/mozilla/TTS): 文本转语音工具。
- [Tacotron2](https://github.com/NVIDIA/tacotron2): 高效文本到语音生成工具。
---
## 深度学习工具
- [pytorch-lightning](https://github.com/Lightning-AI/lightning): 提高 PyTorch 代码可读性和训练效率的框架。
- [ignite](https://github.com/pytorch/ignite): 高级训练工具库。
- [fastai](https://github.com/fastai/fastai): 快速构建深度学习模型的工具。
- [torchmetrics](https://github.com/PyTorchLightning/metrics): 模型评估指标工具。
---
## 图神经网络
- [PyG](https://github.com/pyg-team/pytorch_geometric): 图深度学习扩展库。
- [DGL](https://github.com/dmlc/dgl): 深度图学习框架。
---
## 推荐教程
- [PyTorch 官方教程](https://pytorch.org/tutorials/): 官方提供的从入门到进阶的教程。
- [pytorch-tutorial](https://github.com/yunjey/pytorch-tutorial): 适合研究人员的 PyTorch 深度学习教程。
- [Deep Learning with PyTorch](https://www.manning.com/books/deep-learning-with-pytorch): 《PyTorch 深度学习》书籍案例。
- [Dive into Deep Learning (PyTorch)](https://github.com/dsgiitr/d2l-pytorch): 《动手学深度学习》PyTorch 版。
---
## 常见论文实现
- [transformers](https://github.com/huggingface/transformers): Attention is All You Need 的实现。
- [pix2pix](https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix): 图像到图像翻译的经典实现。
- [CycleGAN](https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix): 无监督图像到图像翻译。
- [DeepSpeech2](https://github.com/SeanNaren/deepspeech.pytorch): 语音识别模型 DeepSpeech2 的实现。
- [faster-rcnn.pytorch](https://github.com/jwyang/faster-rcnn.pytorch): Faster R-CNN 的 PyTorch 实现。
---
## 实用工具
- [tensorboard-pytorch](https://github.com/lanpa/tensorboard-pytorch): TensorBoard 的 PyTorch 插件。
- [torchsummary](https://github.com/sksq96/pytorch-summary): 模型结构和参数总结工具。
- [torchstat](https://github.com/Swall0w/torchstat): 模型分析工具。
- [pytorch-OpCounter](https://github.com/Lyken17/pytorch-OpCounter): 计算模型的 FLOPs。