Alternatives and similar repositories for vision

Users that are interested in vision are comparing it to the libraries listed below

• Cadene / pretrained-models.pytorch
  Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.
  ☆9,105Updated 3 years ago
• pytorch / examples
  A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
  ☆23,439Updated last month
• lukemelas / EfficientNet-PyTorch
  A PyTorch implementation of EfficientNet
  ☆8,162Updated 3 years ago
• facebookresearch / detr
  End-to-End Object Detection with Transformers
  ☆14,746Updated last year
• NVIDIA / apex
  A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
  ☆8,813Updated this week
• lanpa / tensorboardX
  tensorboard for pytorch (and chainer, mxnet, numpy, ...)
  ☆7,965Updated last month
• albumentations-team / albumentations
  Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125
  ☆15,159Updated 3 months ago
• cocodataset / cocoapi
  COCO API - Dataset @ http://cocodataset.org/
  ☆6,302Updated last year
• vdumoulin / conv_arithmetic
  A technical report on convolution arithmetic in the context of deep learning
  ☆14,503Updated 2 years ago
• facebookresearch / maskrcnn-benchmark
  Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.
  ☆9,377Updated 2 years ago
• jacobgil / pytorch-grad-cam
  Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, I…
  ☆12,224Updated 6 months ago
• huggingface / pytorch-image-models
  The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…
  ☆35,398Updated last week
• pytorch / tutorials
  PyTorch tutorials.
  ☆8,817Updated this week
• jwyang / faster-rcnn.pytorch
  A faster pytorch implementation of faster r-cnn
  ☆7,835Updated 3 years ago
• utkuozbulak / pytorch-cnn-visualizations
  Pytorch implementation of convolutional neural network visualization techniques
  ☆8,116Updated 9 months ago
• facebookresearch / detectron2
  Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.
  ☆33,434Updated last month
• aleju / imgaug
  Image augmentation for machine learning experiments.
  ☆14,680Updated last year
• microsoft / Swin-Transformer
  This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".
  ☆15,249Updated last year
• ultralytics / yolov3
  YOLOv3 in PyTorch > ONNX > CoreML > TFLite
  ☆10,463Updated 2 weeks ago
• google-research / vision_transformer
  ☆11,828Updated 7 months ago
• KaimingHe / deep-residual-networks
  Deep Residual Learning for Image Recognition
  ☆6,636Updated 7 years ago
• open-mmlab / mmcv
  OpenMMLab Computer Vision Foundation
  ☆6,269Updated 5 months ago
• arogozhnikov / einops
  Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
  ☆9,205Updated last month
• Lyken17 / pytorch-OpCounter
  Count the MACs / FLOPs of your PyTorch model.
  ☆5,049Updated last year
• open-mmlab / mmdetection
  OpenMMLab Detection Toolbox and Benchmark
  ☆31,757Updated last year
• kornia / kornia
  🐍 Geometric Computer Vision Library for Spatial AI
  ☆10,782Updated last week
• sksq96 / pytorch-summary
  Model summary in PyTorch similar to `model.summary()` in Keras
  ☆4,061Updated last year
• weiaicunzai / pytorch-cifar100
  Practice on cifar100(ResNet, DenseNet, VGG, GoogleNet, InceptionV3, InceptionV4, Inception-ResNetv2, Xception, Resnet In Resnet, ResNext,…
  ☆4,639Updated last year
• facebookresearch / SlowFast
  PySlowFast: video understanding codebase from FAIR for reproducing state-of-the-art video models.
  ☆7,117Updated 10 months ago
• dmlc / gluon-cv
  Gluon CV Toolkit
  ☆5,910Updated 10 months ago