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

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-db

Refine high-quality datasets and visual AI models

gluoncv

Gluon CV Toolkit

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

torchxrayvision

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

cvat-sdk

Annotate better with CVAT, the industry-leading data engine for machine learning. Used and trusted by teams at any scale, for data of any scale.

efficientnet

Implementation of EfficientNet model. Keras and TensorFlow Keras.

grad-cam

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

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.

opennsfw2

Keras implementation of the Yahoo Open-NSFW model

mmcls

OpenMMLab Pre-training Toolbox and Benchmark

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.

medpy

Medical image processing in Python

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

pixlstash

PixlStash is a Python-based image management, tagging and editing web app leveraging AI tools for tagging, similarity checks, grouping and quality assessment. It has a REST-API and a VUE-based web frontend.