PyTorchの便利な周辺ライブラリのメモ.
CV系
- torchvision: コンピュータビジョンで便利なtransforms, モデル,データセットなど
- pytorch-semantic-segmentation: PyTorchによるセマンティックセグメンテーションモデルの実装
- RoIAlign.pytorch: RoIAlignのPyTorch実装
- pytorch-cnn-finetune: CNNのファインチューニング
- detectorch: PyTorchによるObject Detectionモデルの実装
- Augmentor: データ拡張ライブラリ
- PyTorchCV: PyTorchベースのCV用フレームワーク
- imgclsmob: ImageNet学習モデル集
- medicaltorch: 医療画像処理のためのライブラリ
- PyTorch-YOLOv3: YOLO v3のPyTorch実装
- PyTorch-GAN: PyTorchのGAN実装集
- colorization-pytorch: ColorizationのPyTorch実装集
NLP
- torchtext: NLPで便利なデータおよびデータローダの提供
- pytorch-seq2seq: seq2seqモデルのフレームワーク
- fairseq: Facebook AI Researchによるseq2seq Toolkit
- Open-NMT-py: Neural Machine TranslationのPyTorch実装
- AllenNLP:: オープンソースのNLP研究のためのライブラリ
- PyTorch-NLP: NLPのためのutilsやデータローダなどの提供
- BERT-pytorch: BERTのPyTorch実装
- pytext: PyTorchベースのNLPフレームワーク
Audio
- torchaudio: Audio I/Oやデータローダの提供
- pytorch-wavenet: WaveNetのPyTorch実装
確率的プログラミング
- pyro: PyTorchの確率的プログラミングライブラリ
- https://github.com/probtorch/probtorch: Pyroに影響を受けた確率的プログラミングライブラリ
- pyvarinf: Bayesian Deep Learning methodsライブラリ
可視化
- tensorboardX:TensorboardをPyTorchで使えるようにするライブラリ
- pytorchviz: PyTorchの計算グラフの可視化ライブラリ
- pytorch-summary:
keras.summary()
ライクなモデル情報出力のライブラリ
その他
- ignite: PyTorchのハイレベルAPIラッパー
- pytorch_fft: FFTのPyTorchラッパー
- gpytorch: Gaussian Processライブラリ
- pytorch-caffe-darknet-convert: PyTorch, Caffe, darknet間の変換ライブラリ
- semi-supervised-pytorch: PyTorchの半教師あり学習ライブラリ
- pytorch_cluster: PyTorchのグラフベースのクラスタリングアルゴリズムの実装
- extension-cpp: PyTorchのC++拡張
- jetson-reinforcement: NVIDIA Jetsonで動作する強化学習ライブラリ
- beuty-net: PyTorchのプロジェクトテンプレート
- QNNPACK: モデルの量子化ライブラリ
- pytorch2keras: PyTorchのモデルをKerasモデルに変換
- hessian: hessianのPyTorch実装