Image Classification Python Packages

timm

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

ultralytics

Ultralytics YOLO26, YOLO11, YOLOv8 — object detection, instance segmentation, semantic segmentation, image classification, pose estimation, object tracking

tensorflow-hub

A library for transfer learning by reusing parts of TensorFlow models.

fiftyone-db

Refine high-quality datasets and visual AI models

fiftyone

Refine high-quality datasets and visual AI models

efficientnet

Implementation of EfficientNet model. Keras and TensorFlow Keras.

vit-pytorch

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

cvat-sdk

Computer Vision Annotation Tool (CVAT) is a leading platform for building high-quality visual datasets for vision AI. It offers open-source, cloud, and enterprise products, as well as labeling services, for image, video, and 3D annotation with AI-assisted labeling, quality assurance, team collaboration, analytics, and developer APIs.

label-studio

Label Studio is a multi-type data labeling and annotation tool with standardized output format

clarifai

Experience the power of Clarifai’s AI platform with the python SDK. 🌟 Star to support our work!

torchxrayvision

TorchXRayVision: A library of chest X-ray datasets and models. Classifiers, segmentation, and autoencoders.

grad-cam

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.

gluoncv

Gluon CV Toolkit

opennsfw2

Keras implementation of the Yahoo Open-NSFW model

catalyst

Accelerated deep learning R&D

ipyplot

IPyPlot is a small python package offering fast and efficient plotting of images inside Python Notebooks. It's using IPython with HTML for faster, richer and more interactive way of displaying big numbers of images.

medpy

Medical image processing in Python

ultralytics-opencv-headless

Ultralytics YOLO26, YOLO11, YOLOv8 — object detection, instance segmentation, semantic segmentation, image classification, pose estimation, object tracking

mmcls

OpenMMLab Pre-training Toolbox and Benchmark

labelimg

LabelImg is now part of the Label Studio community. The popular image annotation tool created by Tzutalin is no longer actively being developed, but you can check out Label Studio, the open source data labeling tool for images, text, hypertext, audio, video and time-series data.

remotior-sensus

A Python package that allows for the processing of remote sensing images and GIS data.

mmpretrain

OpenMMLab Pre-training Toolbox and Benchmark

albumentationsx

Next-generation Albumentations: dual-licensed for open-source and commercial use

tf-hub-nightly

A library for transfer learning by reusing parts of TensorFlow models.