PyTorch implementation of EfficientNet V2
Reproduction of EfficientNet V2 architecture as described in EfficientNetV2: Smaller Models and Faster Training by Mingxing Tan, Quoc V. Le with the PyTorch framework.
Models
| Architecture | # Parameters | FLOPs | Top-1 Acc. (%) |
|---|---|---|---|
| EfficientNetV2-S | 22.10M | 8.42G @ 384 | |
| EfficientNetV2-M | 55.30M | 24.74G @ 480 | |
| EfficientNetV2-L | 119.36M | 56.13G @ 480 | |
| EfficientNetV2-XL | 208.96M | 93.41G @ 512 |
Stay tuned for ImageNet pre-trained weights.
Acknowledgement
The implementation is heavily borrowed from HBONet or MobileNetV2, please kindly consider citing the following
@InProceedings{Li_2019_ICCV,
author = {Li, Duo and Zhou, Aojun and Yao, Anbang},
title = {HBONet: Harmonious Bottleneck on Two Orthogonal Dimensions},
booktitle = {The IEEE International Conference on Computer Vision (ICCV)},
month = {Oct},
year = {2019}
}
@InProceedings{Sandler_2018_CVPR,
author = {Sandler, Mark and Howard, Andrew and Zhu, Menglong and Zhmoginov, Andrey and Chen, Liang-Chieh},
title = {MobileNetV2: Inverted Residuals and Linear Bottlenecks},
booktitle = {The IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2018}
}
The official TensorFlow implementation by @mingxingtan.
