lukemelas / EfficientNet-PyTorch
A PyTorch implementation of EfficientNet
☆7,854Updated 2 years ago
Related projects: ⓘ
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,010Updated 2 years ago
- Pytorch implementation of convolutional neural network visualization techniques☆7,814Updated last year
- End-to-End Object Detection with Transformers☆13,367Updated 6 months ago
- A faster pytorch implementation of faster r-cnn☆7,656Updated 2 years ago
- The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.☆5,197Updated 2 years ago
- Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.☆9,393Updated this week
- Sandbox for training deep learning networks☆2,950Updated last week
- Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, I…☆10,218Updated 3 weeks ago
- ResNeSt: Split-Attention Networks☆3,223Updated last year
- Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125☆14,031Updated this week
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆31,479Updated this week
- Reference models and tools for Cloud TPUs.☆5,210Updated 3 weeks ago
- PyTorch implementation of the U-Net for image semantic segmentation with high quality images☆8,936Updated last month
- Datasets, Transforms and Models specific to Computer Vision☆15,992Updated this week
- Google Brain AutoML☆6,208Updated 5 months ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,303Updated 2 weeks ago
- PySlowFast: video understanding codebase from FAIR for reproducing state-of-the-art video models.☆6,510Updated last month
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,001Updated 6 months ago
- Image augmentation for machine learning experiments.☆14,343Updated last month
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,852Updated 2 weeks ago
- Minimal PyTorch implementation of YOLOv3☆7,309Updated 11 months ago
- OpenMMLab Computer Vision Foundation☆5,822Updated 2 weeks ago
- Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.☆9,286Updated last year
- Most popular metrics used to evaluate object detection algorithms.☆4,927Updated 2 weeks ago
- Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.☆3,986Updated 2 months ago
- Object detection, 3D detection, and pose estimation using center point detection:☆7,245Updated last year
- ☆10,072Updated 3 months ago
- Count the MACs / FLOPs of your PyTorch model.☆4,832Updated 2 months 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
- This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".☆13,602Updated last month