Skip to content

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

License

Notifications You must be signed in to change notification settings

narumiruna/efficientnet-pytorch

Repository files navigation

EfficientNet

https://arxiv.org/abs/1905.11946

Prerequisites

  • Ubuntu
  • Python 3
    • torch 1.0.1
    • torchvision 0.2.2.post3
    • tqdm
    • mlconfig

Usage

Torch Hub

model = torch.hub.load('narumiruna/efficientnet-pytorch', 'efficientnet_b0', pretrained=True)

Train

$ python train.py -c /path/to/config

Evaluate

$ python evaluate.py --arch efficientnet_b0 -r /path/to/dataset

Pretrained models

Source: https://github.com/tensorflow/tpu/tree/master/models/official/efficientnet

Model Name Top-1 Accuracy
efficientnet_b0 76.52%
efficientnet_b1 77.80%
efficientnet_b2 78.83%
efficientnet_b3 80.19%
efficientnet_b4 82.27%
efficientnet_b5 83.11%
efficientnet_b6
efficientnet_b7

References

About

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

Topics

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages