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 YOLO27, 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

cvat-sdk

CVAT is a leading data annotation platform for image, video, audio, and 3D datasets. It offers open-source, cloud, and enterprise products, as well as labeling services, with AI-assisted labeling, quality assurance, team collaboration, analytics, and developer APIs.

albumentationsx

Image augmentation for computer vision. AGPL-3.0-only or commercial licensing.

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

medpy

Medical image processing in Python

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.

label-studio

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

ultralytics-opencv-headless

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

opennsfw2

Keras implementation of the Yahoo Open-NSFW model

efficientnet

Implementation of EfficientNet model. Keras and TensorFlow Keras.

mmcls

OpenMMLab Pre-training Toolbox and Benchmark

clarifai

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

tf-hub-nightly

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

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.

mmpretrain

OpenMMLab Pre-training Toolbox and Benchmark

torchxrayvision

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

pixlstash

PixlStash helps you find things in an image library that's gotten out of hand. It imports and tags your images automatically, then lets you search by content or face, sort the keepers from the rubbish, and serve data to other tools (like ComfyUI) with a REST API. Use the desktop version or run it headless as a server.

remotior-sensus

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

cvat-cli

CVAT is a leading data annotation platform for image, video, audio, and 3D datasets. It offers open-source, cloud, and enterprise products, as well as labeling services, with AI-assisted labeling, quality assurance, team collaboration, analytics, and developer APIs.