lukemelas / EfficientNet-PyTorch
A PyTorch implementation of EfficientNet
☆8,031Updated 2 years ago
Alternatives and similar repositories for EfficientNet-PyTorch:
Users that are interested in EfficientNet-PyTorch are comparing it to the libraries listed below
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,057Updated 2 years ago
- ResNeSt: Split-Attention Networks☆3,249Updated 2 years ago
- Reference models and tools for Cloud TPUs.☆5,223Updated 2 months ago
- Sandbox for training deep learning networks☆2,993Updated 5 months ago
- Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.☆4,150Updated 3 months ago
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆33,320Updated last week
- Count the MACs / FLOPs of your PyTorch model.☆4,962Updated 7 months ago
- The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.☆5,224Updated 3 years ago
- Code release for ConvNeXt model☆5,889Updated 2 years ago
- Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125☆14,641Updated this week
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,033Updated last year
- Codebase for Image Classification Research, written in PyTorch.☆2,149Updated 11 months ago
- Google Brain AutoML☆6,315Updated this week
- RepVGG: Making VGG-style ConvNets Great Again☆3,374Updated 2 years ago
- Pytorch implementation of convolutional neural network visualization techniques☆7,969Updated 2 months ago
- OpenMMLab Computer Vision Foundation☆6,025Updated this week
- Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, I…☆11,205Updated last month
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,917Updated last month
- A faster pytorch implementation of faster r-cnn☆7,752Updated 2 years ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,566Updated this week
- End-to-End Object Detection with Transformers☆14,036Updated 11 months ago
- Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Py…☆21,906Updated last month
- This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".☆14,363Updated 7 months ago
- Image augmentation for machine learning experiments.☆14,520Updated 7 months ago
- Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.☆10,102Updated this week
- Object detection, 3D detection, and pose estimation using center point detection:☆7,364Updated 2 years ago
- Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.☆9,331Updated 2 years ago
- Implementation of EfficientNet model. Keras and TensorFlow Keras.☆2,073Updated last year
- Pytorch implementation for Semantic Segmentation/Scene Parsing on MIT ADE20K dataset☆4,991Updated last year
- The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.☆6,096Updated 2 months ago