qubvel/efficientnet
A Keras implementation of EfficientNet, a deep learning model for image classification on ImageNet and transfer learning datasets.

Velocity · 7d
+0.8
★ / day
Trend
→steady
star history
This repository provides a Keras and TensorFlow Keras reimplementation of EfficientNet, a lightweight convolutional neural network architecture. It offers pretrained models from EfficientNet-B0 to B7 that achieve high accuracy on ImageNet and other transfer learning datasets with fewer parameters and FLOPS. The implementation relies on AutoML and compound scaling methods for model optimization.