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,119Updated 3 years ago
• huggingface / pytorch-image-models
  The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…
  ☆36,313Updated last week
• albumentations-team / albumentations
  Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125
  ☆15,269Updated 7 months ago
• open-mmlab / mmcv
  OpenMMLab Computer Vision Foundation
  ☆6,403Updated last week
• facebookresearch / maskrcnn-benchmark
  Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.
  ☆9,388Updated 2 years ago
• facebookresearch / detr
  End-to-End Object Detection with Transformers
  ☆15,098Updated last year
• lukemelas / EfficientNet-PyTorch
  A PyTorch implementation of EfficientNet
  ☆8,214Updated 3 years ago
• lanpa / tensorboardX
  tensorboard for pytorch (and chainer, mxnet, numpy, ...)
  ☆7,987Updated 3 months ago
• open-mmlab / mmdetection
  OpenMMLab Detection Toolbox and Benchmark
  ☆32,378Updated last year
• NVIDIA / apex
  A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
  ☆8,908Updated last week
• pytorch / examples
  A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
  ☆23,724Updated 5 months ago
• aleju / imgaug
  Image augmentation for machine learning experiments.
  ☆14,727Updated last year
• facebookresearch / detectron2
  Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.
  ☆34,046Updated 2 months ago
• kornia / kornia
  🐍 Geometric Computer Vision Library for Spatial AI
  ☆11,064Updated this week
• utkuozbulak / pytorch-cnn-visualizations
  Pytorch implementation of convolutional neural network visualization techniques
  ☆8,197Updated last year
• google-research / vision_transformer
  ☆12,275Updated last week
• facebookresearch / ConvNeXt
  Code release for ConvNeXt model
  ☆6,286Updated 3 years ago
• jwyang / faster-rcnn.pytorch
  A faster pytorch implementation of faster r-cnn
  ☆7,865Updated 3 years ago
• jacobgil / pytorch-grad-cam
  Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, I…
  ☆12,575Updated 10 months ago
• cocodataset / cocoapi
  COCO API - Dataset @ http://cocodataset.org/
  ☆6,357Updated last year
• microsoft / Swin-Transformer
  This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".
  ☆15,700Updated last year
• kuangliu / pytorch-cifar
  95.47% on CIFAR10 with PyTorch
  ☆6,356Updated 2 years ago
• ultralytics / yolov3
  YOLOv3 in PyTorch > ONNX > CoreML > TFLite
  ☆10,552Updated this week
• pytorch / pytorch
  Tensors and Dynamic neural networks in Python with strong GPU acceleration
  ☆97,130Updated this week
• hoya012 / deep_learning_object_detection
  A paper list of object detection using deep learning.
  ☆11,437Updated last year
• milesial / Pytorch-UNet
  PyTorch implementation of the U-Net for image semantic segmentation with high quality images
  ☆11,177Updated last year
• qubvel-org / segmentation_models.pytorch
  Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
  ☆11,312Updated last month
• google / automl
  Google Brain AutoML
  ☆6,452Updated 11 months ago
• Lyken17 / pytorch-OpCounter
  Count the MACs / FLOPs of your PyTorch model.
  ☆5,077Updated last year
• pytorch / tutorials
  PyTorch tutorials.
  ☆9,019Updated last week