PyRank
  • Insights
  • PyPI
  • GitHub
  • Search
  • Compare
  • Advisories
  • Ecosystem
  • About

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.4M 37K 5K
ultralytics
ultralytics

Ultralytics YOLO ๐Ÿš€

10.5M 57K 11K
tensorflow
tensorflow-hub

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

2M 4K 2K
voxel51
fiftyone-db

Refine high-quality datasets and visual AI models

170K 11K 761
Clarifai
clarifai

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

152K 44 8
voxel51
fiftyone

Refine high-quality datasets and visual AI models

136K 11K 761
lucidrains
vit-pytorch

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

118K 25K 3K
HumanSignal
label-studio

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

107K 27K 4K
qubvel
efficientnet

Implementation of EfficientNet model. Keras and TensorFlow Keras.

96K 2K 463
cvat-ai
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.

89K 16K 4K
mlmed
torchxrayvision

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

85K 1K 249
jacobgil
grad-cam

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

77K 13K 2K
dmlc
gluoncv

Gluon CV Toolkit

65K 6K 1K
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.

59K 428 40
tensorflow
tf-hub-nightly

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

55K 4K 2K
bhky
opennsfw2

Keras implementation of the Yahoo Open-NSFW model

52K 492 69
loli
medpy

Medical image processing in Python

35K 618 145
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.

29K 25K 7K
open-mmlab
mmcls

OpenMMLab Pre-training Toolbox and Benchmark

27K 4K 1K
catalyst-team
catalyst

Accelerated deep learning R&D

22K 3K 400
open-mmlab
mmpretrain

OpenMMLab Pre-training Toolbox and Benchmark

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

Ultralytics YOLO ๐Ÿš€

20K 57K 11K
albumentations-team
albumentationsx

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

18K 325 28
cleanlab
cleanvision

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

10K 1K 82
    • Data from PyPI, GitHub, ClickHouse, and BigQuery