- this paper
- another paper
- pytorch vision: Datasets, Transforms and Models specific to Computer Vision.
- Fine-tune pretrained Convolutional Neural Networks with PyTorch
- pytorch-image-models
- detectron2: Detectron2 is FAIR's next-generation research platform for object detection and segmentation. facebookresearch
- MMDetection: MMDetection is an open source object detection toolbox, a part of the OpenMMLab project.
- ClassyVision : An end-to-end PyTorch framework for image and video classification. facebookresearch
- OpenFacePytorch: PyTorch module to use OpenFace's nn4.small2.v1.t7 model
- img_classification_pk_pytorch: Quickly comparing your image classification models with the state-of-the-art models (such as DenseNet, ResNet, ...)
- SparseConvNet: Submanifold sparse convolutional networks.
- Convolution_LSTM_pytorch: A multi-layer convolution LSTM module
- face-alignment: 🔥 2D and 3D Face alignment library build using pytorch adrianbulat.com
- pytorch-semantic-segmentation: PyTorch for Semantic Segmentation.
- RoIAlign.pytorch: This is a PyTorch version of RoIAlign. This implementation is based on crop_and_resize and supports both forward and backward on CPU and GPU.
- pytorch-cnn-finetune: Fine-tune pretrained Convolutional Neural Networks with PyTorch.
- detectorch: Detectorch - detectron for PyTorch
- Augmentor: Image augmentation library in Python for machine learning. http://augmentor.readthedocs.io
- s2cnn: This library contains a PyTorch implementation of the SO(3) equivariant CNNs for spherical signals (e.g. omnidirectional cameras, signals on the globe)
- TorchCV: A PyTorch-Based Framework for Deep Learning in Computer Vision.
- maskrcnn-benchmark: Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.
- image-classification-mobile: Collection of classification models pretrained on the ImageNet-1K.
- medicaltorch: A medical imaging framework for Pytorch http://medicaltorch.readthedocs.io
- albumentations: Fast image augmentation library.
- kornia: Differentiable computer vision library.
- pytorch-text-recognition: Text recognition combo - CRAFT + CRNN.
- facenet-pytorch: Pretrained Pytorch face detection and recognition models ported from davidsandberg/facenet.
- detectron2: Detectron2 is FAIR's next-generation research platform for object detection and segmentation.
- vedaseg: A semantic segmentation framework by pyotrch
- ClassyVision: An end-to-end PyTorch framework for image and video classification.
- detecto:Computer vision in Python with less than 10 lines of code
- pytorch3d: PyTorch3D is FAIR's library of reusable components for deep learning with 3D data pytorch3d.org
- MMDetection: MMDetection is an open source object detection toolbox, a part of the OpenMMLab project.
- neural-dream: A PyTorch implementation of the DeepDream algorithm. Creates dream-like hallucinogenic visuals.
- FlashTorch: Visualization toolkit for neural networks in PyTorch!
- Lucent: Tensorflow and OpenAI Clarity's Lucid adapted for PyTorch.
- MMDetection3D: MMDetection3D is OpenMMLab's next-generation platform for general 3D object detection, a part of the OpenMMLab project.
- MMSegmentation: MMSegmentation is a semantic segmentation toolbox and benchmark, a part of the OpenMMLab project.
- MMEditing: MMEditing is a image and video editing toolbox, a part of the OpenMMLab project.
- MMAction2: MMAction2 is OpenMMLab's next generation action understanding toolbox and benchmark, a part of the OpenMMLab project.
- MMPose: MMPose is a pose estimation toolbox and benchmark, a part of the OpenMMLab project.
- lightly - Lightly is a computer vision framework for self-supervised learning.