PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more

Google Brain AutoML

NanoDet-Plus⚡Super fast and lightweight anchor-free object detection model. 🔥Only 980 KB(int8) / 1.8MB (fp16) and run 97FPS on cellphone🔥

The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

Implementation of EfficientNet model. Keras and TensorFlow Keras.

Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS

A PyTorch impl of EfficientDet faithful to the original Google impl w/ ported weights

Implementation EfficientDet: Scalable and Efficient Object Detection in PyTorch

Bumble's Private Detector - a pretrained model for detecting lewd images

High level network definitions with pre-trained weights in TensorFlow

MEAL V2: Boosting Vanilla ResNet-50 to 80%+ Top-1 Accuracy on ImageNet without Tricks

(Pretrained weights provided) EfficientDet: Scalable and Efficient Object Detection implementation by Signatrix GmbH

More readable and flexible yolov5 with more backbone(gcn, resnet, shufflenet, moblienet, efficientnet, hrnet, swin-transformer, etc) and (cbam，dcn and so on), and tensorrt

A tensorflow2 implementation of some basic CNNs(MobileNetV1/V2/V3, EfficientNet, ResNeXt, InceptionV4, InceptionResNetV1/V2, SENet, SqueezeNet, DenseNet, ShuffleNetV2, ResNet).

A list of popular deep learning models related to classification, segmentation and detection problems

face recognition training project(pytorch)

PyTorch implementation of EfficientNetV2 family

📸 PyTorch implementation of MobileNetV3 for real-time semantic segmentation, with pretrained weights & state-of-the-art performance

A PyTorch implementation of " EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks."