lukemelas/EfficientNet-PyTorch

Health v2 · maintenance only

Adoption not part of the score

8222 stars · 1535 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-29, confidence not recorded

A PyTorch implementation of the EfficientNet convolutional neural network family with pretrained ImageNet weights. It provides a simple pip-installable API for loading models for classification and feature extraction.

Use cases

• load pretrained efficientnet in pytorch
• image classification with efficientnet
• extract features from images with a cnn
• transfer learning backbone for vision models
• run imagenet classification inference

When to choose

• you need EfficientNet specifically in PyTorch with pretrained weights
• you want a lightweight pip-installable vision backbone for feature extraction or fine-tuning

When to avoid

• you need newer architectures like EfficientNetV2 or timm's broader model zoo
• you use TensorFlow or another framework
• you need actively maintained code with recent fixes

Facets

library · maturity maintenance

machine-learning deep-learning image-processing deep-learning computer-vision image-processing python pytorch efficientnet pretrained-models imagenet feature-extraction transfer-learning

Member repositories

For agents

markdown · JSON · MCP: product_card(name="lukemelas/EfficientNet-PyTorch")