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/configEvaluate
$ python evaluate.py --arch efficientnet_b0 -r /path/to/datasetPretrained 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 |