pytorch / visionLinks
Datasets, Transforms and Models specific to Computer Vision
☆17,056Updated this week
Alternatives and similar repositories for vision
Users that are interested in vision are comparing it to the libraries listed below
Sorting:
- OpenMMLab Computer Vision Foundation☆6,216Updated 3 months ago
- Image augmentation for machine learning experiments.☆14,643Updated last year
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,763Updated this week
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,098Updated 3 years ago
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆23,309Updated last week
- End-to-End Object Detection with Transformers☆14,590Updated last year
- OpenMMLab Detection Toolbox and Benchmark☆31,520Updated 11 months ago
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,964Updated last month
- Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125☆15,105Updated last month
- A PyTorch implementation of EfficientNet☆8,142Updated 3 years ago
- COCO API - Dataset @ http://cocodataset.org/☆6,282Updated last year
- This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".☆15,080Updated last year
- Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.☆9,370Updated 2 years ago
- PyTorch tutorials.☆8,730Updated this week
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆35,004Updated this week
- ☆11,649Updated 5 months ago
- A faster pytorch implementation of faster r-cnn☆7,823Updated 3 years ago
- Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.☆32,518Updated last month
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,052Updated last year
- Reference models and tools for Cloud TPUs.☆5,252Updated last month
- Count the MACs / FLOPs of your PyTorch model.☆5,029Updated last year
- Pytorch implementation of convolutional neural network visualization techniques☆8,073Updated 7 months ago
- Code release for ConvNeXt model☆6,085Updated 2 years ago
- Squeeze-and-Excitation Networks☆3,545Updated 6 years ago
- A PyTorch Implementation of Single Shot MultiBox Detector☆5,213Updated 3 years ago
- A simplified implemention of Faster R-CNN that replicate performance from origin paper☆4,021Updated 4 years ago
- 🐍 Geometric Computer Vision Library for Spatial AI☆10,668Updated this week
- Tensors and Dynamic neural networks in Python with strong GPU acceleration☆92,272Updated this week
- Object detection, 3D detection, and pose estimation using center point detection:☆7,453Updated 2 years ago
- Image Polygonal Annotation with Python (polygon, rectangle, circle, line, point and image-level flag annotation).☆14,859Updated last month