pytorch / vision

Related projects ⓘ

Alternatives and complementary repositories for vision

• pytorch / examples
  A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
  ☆22,439Updated last week
• Cadene / pretrained-models.pytorch
  Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.
  ☆9,038Updated 2 years ago
• facebookresearch / detr
  End-to-End Object Detection with Transformers
  ☆13,626Updated 8 months ago
• huggingface / pytorch-image-models
  The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…
  ☆32,290Updated this week
• pytorch / tutorials
  PyTorch tutorials.
  ☆8,238Updated this week
• lukemelas / EfficientNet-PyTorch
  A PyTorch implementation of EfficientNet
  ☆7,926Updated 2 years ago
• facebookresearch / maskrcnn-benchmark
  Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.
  ☆9,303Updated last year
• albumentations-team / albumentations
  Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125
  ☆14,273Updated this week
• lanpa / tensorboardX
  tensorboard for pytorch (and chainer, mxnet, numpy, ...)
  ☆7,871Updated this week
• NVIDIA / apex
  A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
  ☆8,415Updated 2 weeks ago
• lucidrains / vit-pytorch
  Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Py…
  ☆20,652Updated last week
• open-mmlab / mmcv
  OpenMMLab Computer Vision Foundation
  ☆5,907Updated this week
• jwyang / faster-rcnn.pytorch
  A faster pytorch implementation of faster r-cnn
  ☆7,695Updated 2 years ago
• utkuozbulak / pytorch-cnn-visualizations
  Pytorch implementation of convolutional neural network visualization techniques
  ☆7,869Updated 2 years ago
• microsoft / Swin-Transformer
  This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".
  ☆13,929Updated 3 months ago
• google-research / vision_transformer
  ☆10,467Updated 6 months ago
• kuangliu / pytorch-cifar
  95.47% on CIFAR10 with PyTorch
  ☆5,992Updated last year
• vdumoulin / conv_arithmetic
  A technical report on convolution arithmetic in the context of deep learning
  ☆14,070Updated last year
• sksq96 / pytorch-summary
  Model summary in PyTorch similar to `model.summary()` in Keras
  ☆4,020Updated 8 months ago
• yunjey / pytorch-tutorial
  PyTorch Tutorial for Deep Learning Researchers
  ☆30,271Updated last year
• facebookresearch / ConvNeXt
  Code release for ConvNeXt model
  ☆5,774Updated last year
• ultralytics / yolov3
  YOLOv3 in PyTorch > ONNX > CoreML > TFLite
  ☆10,233Updated this week
• cocodataset / cocoapi
  COCO API - Dataset @ http://cocodataset.org/
  ☆6,109Updated 7 months ago
• weiaicunzai / pytorch-cifar100
  Practice on cifar100(ResNet, DenseNet, VGG, GoogleNet, InceptionV3, InceptionV4, Inception-ResNetv2, Xception, Resnet In Resnet, ResNext,…
  ☆4,294Updated 4 months ago
• milesial / Pytorch-UNet
  PyTorch implementation of the U-Net for image semantic segmentation with high quality images
  ☆9,269Updated 3 months ago
• kornia / kornia
  Geometric Computer Vision Library for Spatial AI
  ☆9,978Updated this week
• dmlc / gluon-cv
  Gluon CV Toolkit
  ☆5,835Updated 7 months ago
• CSAILVision / semantic-segmentation-pytorch
  Pytorch implementation for Semantic Segmentation/Scene Parsing on MIT ADE20K dataset
  ☆4,948Updated 10 months ago
• hoya012 / deep_learning_object_detection
  A paper list of object detection using deep learning.
  ☆11,319Updated 9 months ago
• aleju / imgaug
  Image augmentation for machine learning experiments.
  ☆14,418Updated 3 months ago