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 YOLO 🚀

tensorflow-hub

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

fiftyone-db

Refine high-quality datasets and visual AI models

clarifai

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

fiftyone

Refine high-quality datasets and visual AI models

vit-pytorch

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

label-studio

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

efficientnet

Implementation of EfficientNet model. Keras and TensorFlow Keras.

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.

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

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.

tf-hub-nightly

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

opennsfw2

Keras implementation of the Yahoo Open-NSFW model

medpy

Medical image processing in Python

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.

mmcls

OpenMMLab Pre-training Toolbox and Benchmark

catalyst

Accelerated deep learning R&D

mmpretrain

OpenMMLab Pre-training Toolbox and Benchmark

ultralytics-opencv-headless

Ultralytics YOLO 🚀

albumentationsx

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

cleanvision

Automatically find issues in image datasets and practice data-centric computer vision.