lukemelas/EfficientNet-PyTorch
A PyTorch implementation of EfficientNet, a convolutional neural network architecture for image classification.

Velocity · 7d
+3.2
★ / day
Trend
→steady
star history
This repository provides a PyTorch implementation of EfficientNet, a family of efficient and scalable convolutional neural network architectures for image classification. It includes pretrained models on ImageNet with support for feature extraction. The implementation supports loading various EfficientNet variants including the original EfficientNet and the newer EfficientNetV2, with optional adversarial training (advprop) preprocessing.