PyPI Stats
  • Insights
  • PyPI
  • GitHub
  • Search
  • Compare
  • Advisories
  • Ecosystem
  • About
Home

Image Classification Python Packages

Python packages with the GitHub topic image-classification. Sorted by relevance, with stars and monthly downloads.
huggingface
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

13.3M 37K 5K
ultralytics
ultralytics

Ultralytics YOLO ๐Ÿš€

10.7M 57K 11K
tensorflow
tensorflow-hub

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

2M 4K 2K
voxel51
fiftyone

Refine high-quality datasets and visual AI models

178K 11K 752
Clarifai
clarifai

Experience the power of Clarifaiโ€™s AI platform with the python SDK. ๐ŸŒŸ Star to support our work!

176K 42 8
voxel51
fiftyone-db

Refine high-quality datasets and visual AI models

171K 11K 752
dmlc
gluoncv

Gluon CV Toolkit

168K 6K 1K
lucidrains
vit-pytorch

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

111K 25K 3K
HumanSignal
label-studio

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

111K 27K 4K
mlmed
torchxrayvision

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

95K 1K 248
cvat-ai
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.

85K 16K 4K
qubvel
efficientnet

Implementation of EfficientNet model. Keras and TensorFlow Keras.

84K 2K 463
jacobgil
grad-cam

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

82K 13K 2K
karolzak
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.

58K 428 40
bhky
opennsfw2

Keras implementation of the Yahoo Open-NSFW model

52K 489 70
open-mmlab
mmcls

OpenMMLab Pre-training Toolbox and Benchmark

50K 4K 1K
tensorflow
tf-hub-nightly

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

47K 4K 2K
tzutalin
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.

30K 25K 7K
loli
medpy

Medical image processing in Python

30K 617 145
catalyst-team
catalyst

Accelerated deep learning R&D

27K 3K 400
open-mmlab
mmpretrain

OpenMMLab Pre-training Toolbox and Benchmark

21K 4K 1K
ultralytics
ultralytics-opencv-headless

Ultralytics YOLO ๐Ÿš€

19K 57K 11K
albumentations-team
albumentationsx

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

16K 315 28
pikselkroken
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.

10K 40 4
    • Data from PyPI, GitHub, ClickHouse, and BigQuery