← all repositories

lukemelas/EfficientNet-PyTorch

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

8.2k stars Python Computer VisionML Frameworks
EfficientNet-PyTorch
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.