qubvel/classification_models

Health v2 · maintenance only

Adoption not part of the score

1231 stars · 308 forks observed · 2026-08-28

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

A Keras/TensorFlow Keras library providing ImageNet-pretrained image classification model architectures such as ResNet, SE-ResNet, DenseNet, MobileNet, and Inception. It is commonly used to obtain pretrained backbones for transfer learning and computer vision tasks.

Use cases

• load imagenet pretrained resnet in keras
• get a pretrained backbone for image classification
• transfer learning with keras classification models
• compare imagenet accuracy of resnet densenet mobilenet
• use se-resnet or resnext weights in tensorflow keras
• feature extractor backbone for segmentation model

When to choose

• you need ImageNet-pretrained classification backbones in Keras or tf.keras
• you want architectures like SE-ResNet or ResNeXt not available in keras-applications
• you are doing transfer learning or building custom vision models on top of backbones

When to avoid

• you use PyTorch instead of Keras/TensorFlow
• you need the latest architectures or actively maintained releases
• you need models beyond ImageNet classification, such as detection heads

Facets

library · maturity maintenance

machine-learning deep-learning image-processing deep-learning computer-vision image-processing python keras tensorflow imagenet pretrained-weights resnet densenet mobilenet image-classification backbone

Member repositories

For agents

markdown · JSON · MCP: product_card(name="qubvel/classification_models")