Alternatives and similar repositories for vision

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

• albumentations-team / albumentations
  Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125
  ☆15,228Updated 5 months ago
• pytorch / examples
  A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
  ☆23,618Updated 3 months ago
• huggingface / pytorch-image-models
  The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…
  ☆36,035Updated this week
• Cadene / pretrained-models.pytorch
  Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.
  ☆9,117Updated 3 years ago
• facebookresearch / detr
  End-to-End Object Detection with Transformers
  ☆14,985Updated last year
• NVIDIA / apex
  A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
  ☆8,873Updated this week
• microsoft / Swin-Transformer
  This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".
  ☆15,556Updated last year
• lukemelas / EfficientNet-PyTorch
  A PyTorch implementation of EfficientNet
  ☆8,204Updated 3 years ago
• lanpa / tensorboardX
  tensorboard for pytorch (and chainer, mxnet, numpy, ...)
  ☆7,979Updated last month
• utkuozbulak / pytorch-cnn-visualizations
  Pytorch implementation of convolutional neural network visualization techniques
  ☆8,176Updated 11 months ago
• open-mmlab / mmcv
  OpenMMLab Computer Vision Foundation
  ☆6,348Updated last week
• aleju / imgaug
  Image augmentation for machine learning experiments.
  ☆14,718Updated last year
• jacobgil / pytorch-grad-cam
  Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, I…
  ☆12,451Updated 8 months ago
• vdumoulin / conv_arithmetic
  A technical report on convolution arithmetic in the context of deep learning
  ☆14,567Updated 2 years ago
• open-mmlab / mmdetection
  OpenMMLab Detection Toolbox and Benchmark
  ☆32,189Updated last year
• pytorch / tutorials
  PyTorch tutorials.
  ☆8,931Updated this week
• jwyang / faster-rcnn.pytorch
  A faster pytorch implementation of faster r-cnn
  ☆7,853Updated 3 years ago
• google / automl
  Google Brain AutoML
  ☆6,436Updated 9 months ago
• Lightning-AI / pytorch-lightning
  Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
  ☆30,586Updated this week
• facebookresearch / maskrcnn-benchmark
  Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.
  ☆9,388Updated 2 years ago
• weiaicunzai / pytorch-cifar100
  Practice on cifar100(ResNet, DenseNet, VGG, GoogleNet, InceptionV3, InceptionV4, Inception-ResNetv2, Xception, Resnet In Resnet, ResNext,…
  ☆4,723Updated last year
• ultralytics / yolov3
  YOLOv3 in PyTorch > ONNX > CoreML > TFLite
  ☆10,526Updated this week
• milesial / Pytorch-UNet
  PyTorch implementation of the U-Net for image semantic segmentation with high quality images
  ☆10,999Updated last year
• kuangliu / pytorch-cifar
  95.47% on CIFAR10 with PyTorch
  ☆6,325Updated 2 years ago
• pytorch / pytorch
  Tensors and Dynamic neural networks in Python with strong GPU acceleration
  ☆95,988Updated this week
• eriklindernoren / PyTorch-YOLOv3
  Minimal PyTorch implementation of YOLOv3
  ☆7,441Updated last year
• open-mmlab / mmsegmentation
  OpenMMLab Semantic Segmentation Toolbox and Benchmark.
  ☆9,503Updated last year
• hoya012 / deep_learning_object_detection
  A paper list of object detection using deep learning.
  ☆11,434Updated last year
• cocodataset / cocoapi
  COCO API - Dataset @ http://cocodataset.org/
  ☆6,340Updated last year
• sksq96 / pytorch-summary
  Model summary in PyTorch similar to `model.summary()` in Keras
  ☆4,066Updated last year