pytorch / vision
Datasets, Transforms and Models specific to Computer Vision
☆15,992Updated this week
Related projects: ⓘ
- A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.☆22,236Updated this week
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆31,479Updated this week
- Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125☆14,031Updated this week
- End-to-End Object Detection with Transformers☆13,367Updated 6 months ago
- OpenMMLab Detection Toolbox and Benchmark☆29,098Updated 3 weeks ago
- Image augmentation for machine learning experiments.☆14,343Updated last month
- PyTorch tutorials.☆8,102Updated this week
- A PyTorch implementation of EfficientNet☆7,854Updated 2 years ago
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,010Updated 2 years ago
- PyTorch Tutorial for Deep Learning Researchers☆29,835Updated last year
- A technical report on convolution arithmetic in the context of deep learning☆13,967Updated last year
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,303Updated 2 weeks ago
- Pytorch implementation of convolutional neural network visualization techniques☆7,814Updated last year
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,852Updated 2 weeks ago
- Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Py…☆19,651Updated 3 weeks ago
- Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.☆29,989Updated last week
- PyTorch implementations of Generative Adversarial Networks.☆16,230Updated 3 months ago
- Latex code for making neural networks diagrams☆21,827Updated last year
- A comprehensive list of pytorch related content on github,such as different models,implementations,helper libraries,tutorials etc.☆15,313Updated 7 months ago
- ☆10,072Updated 3 months ago
- This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".☆13,602Updated last month
- Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.☆9,286Updated last year
- A paper list of object detection using deep learning.☆11,287Updated 7 months ago
- Library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.☆15,324Updated last year
- Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, I…☆10,218Updated 2 weeks ago
- A list of all named GANs!☆14,216Updated 11 months ago
- Mask R-CNN for object detection and instance segmentation on Keras and TensorFlow☆24,533Updated 3 months ago
- Visualizer for neural network, deep learning and machine learning models☆27,558Updated this week
- State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enter…☆13,233Updated last month
- Image Polygonal Annotation with Python (polygon, rectangle, circle, line, point and image-level flag annotation).☆13,171Updated this week