Python image-classification

Open-source Python projects categorized as image-classification

Top 23 Python image-classification Projects

image-classification
  1. ultralytics

    Ultralytics YOLO11 🚀

    Project mention: Show HN: Using YOLO to Detect Office Chairs in 40M Hotel Photos | news.ycombinator.com | 2025-01-25

    They did it on their own computer. https://github.com/ultralytics/ultralytics

  2. InfluxDB

    InfluxDB – Built for High-Performance Time Series Workloads. InfluxDB 3 OSS is now GA. Transform, enrich, and act on time series data directly in the database. Automate critical tasks and eliminate the need to move data externally. Download now.

    InfluxDB logo
  3. pytorch-image-models

    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

    Project mention: This PR content was generated automatically using cover-agent | news.ycombinator.com | 2024-11-19

    Those are some pointless tests.

    E.g. test_activation_stats_functions [1] that just checks that the returned value is a float, and that it can take random numbers as input.

    test_get_state_dict_custom_unwrap [2] is probably supposed to check that custom_unwrap is invoked, but since it doesn't either record being called, or transform its input, the assertions can't actually check that it was called.

    [1] https://github.com/huggingface/pytorch-image-models/pull/233...

    [2] https://github.com/huggingface/pytorch-image-models/pull/233...

  4. vit-pytorch

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

  5. albumentations

    Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125

    Project mention: Albumentations: Fast and flexible image augmentation library | news.ycombinator.com | 2025-02-22
  6. Swin-Transformer

    This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".

  7. pytorch-grad-cam

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

  8. fiftyone

    Refine high-quality datasets and visual AI models

    Project mention: Launch HN: Enhanced Radar (YC W25) – A safety net for air traffic control | news.ycombinator.com | 2025-03-04

    Are there already bird not a bird datasets?

    Procedures for creating "bird on Multispectral plane radar and video" dataset(s):

    Tag birds on the dashcam video with timecoded sensor data and a segmentation and annotation tool.

    Pinch to zoom, auto-edge detect, classification probability, sensor status

    voxel51/fiftyone does segmentation and annotation with video and possibly Multispectral data: https://github.com/voxel51/fiftyone

  9. SaaSHub

    SaaSHub - Software Alternatives and Reviews. SaaSHub helps you find the best software and product alternatives

    SaaSHub logo
  10. InternVL

    [CVPR 2024 Oral] InternVL Family: A Pioneering Open-Source Alternative to GPT-4o. 接近GPT-4o表现的开源多模态对话模型

    Project mention: InternVL3: Unified Multimodal AI Training Outperforms Open-Source Rivals | dev.to | 2025-04-17

    InternVL3 marks a significant advancement in the InternVL model series, implementing a native multimodal pre-training approach that fundamentally transforms how vision-language models learn. Unlike most leading multimodal large language models (MLLMs) that adapt text-only models to handle visual inputs through complex post-hoc alignment, InternVL3 jointly acquires multimodal and linguistic capabilities in a single unified pre-training stage.

  11. gluon-cv

    Gluon CV Toolkit

  12. PaddleClas

    A treasure chest for visual classification and recognition powered by PaddlePaddle

  13. mmpretrain

    OpenMMLab Pre-training Toolbox and Benchmark

  14. hub

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

  15. catalyst

    Accelerated deep learning R&D (by catalyst-team)

  16. autodistill

    Images to inference with no labeling (use foundation models to train supervised models).

    Project mention: Ask HN: Who is hiring? (April 2025) | news.ycombinator.com | 2025-04-01
  17. ailia-models

    The collection of pre-trained, state-of-the-art AI models for ailia SDK

  18. sparseml

    Libraries for applying sparsification recipes to neural networks with a few lines of code, enabling faster and smaller models

  19. efficientnet

    Implementation of EfficientNet model. Keras and TensorFlow Keras.

  20. fastdup

    fastdup is a powerful, free tool designed to rapidly generate valuable insights from image and video datasets. It helps enhance the quality of both images and labels, while significantly reducing data operation costs, all with unmatched scalability.

    Project mention: AIM Weekly 17 June 2024 | dev.to | 2024-06-17
  21. pytorch-toolbelt

    PyTorch extensions for fast R&D prototyping and Kaggle farming

  22. Unsupervised-Classification

    SCAN: Learning to Classify Images without Labels, incl. SimCLR. [ECCV 2020]

  23. private-detector

    Bumble's Private Detector - a pretrained model for detecting lewd images

  24. poolformer

    PoolFormer: MetaFormer Is Actually What You Need for Vision (CVPR 2022 Oral)

  25. involution

    [CVPR 2021] Involution: Inverting the Inherence of Convolution for Visual Recognition, a brand new neural operator

  26. SaaSHub

    SaaSHub - Software Alternatives and Reviews. SaaSHub helps you find the best software and product alternatives

    SaaSHub logo
NOTE: The open source projects on this list are ordered by number of github stars. The number of mentions indicates repo mentiontions in the last 12 Months or since we started tracking (Dec 2020).

Python image-classification discussion

Log in or Post with

Python image-classification related posts

  • Show HN: Local, automatic, image keywords, captions using metadata for storage

    1 project | news.ycombinator.com | 1 Apr 2025
  • Alternatives to Cosine Similarity

    1 project | news.ycombinator.com | 8 Oct 2024
  • I made a social media app

    1 project | /r/webdev | 8 Dec 2023
  • Samsung expected to report 80% profit plunge as losses mount at chip business

    3 projects | news.ycombinator.com | 10 Oct 2023
  • Is it easier to go from Pytorch to TF and Keras than the other way around?

    1 project | /r/pytorch | 13 May 2023
  • Swin Transformer: Hierarchical Vision Transformer Using Shifted Windows

    2 projects | news.ycombinator.com | 9 Apr 2023
  • Swin Transformer: Hierarchical Vision Transformer using Shifted Windows

    1 project | /r/neuralnetworks | 9 Apr 2023
  • A note from our sponsor - SaaSHub
    www.saashub.com | 12 May 2025
    SaaSHub helps you find the best software and product alternatives Learn more →

Index

What are some of the best open-source image-classification projects in Python? This list will help you:

# Project Stars
1 ultralytics 40,251
2 pytorch-image-models 34,064
3 vit-pytorch 22,795
4 albumentations 14,908
5 Swin-Transformer 14,746
6 pytorch-grad-cam 11,612
7 fiftyone 9,461
8 InternVL 8,015
9 gluon-cv 5,882
10 PaddleClas 5,644
11 mmpretrain 3,649
12 hub 3,496
13 catalyst 3,336
14 autodistill 2,252
15 ailia-models 2,191
16 sparseml 2,131
17 efficientnet 2,083
18 fastdup 1,680
19 pytorch-toolbelt 1,541
20 Unsupervised-Classification 1,418
21 private-detector 1,330
22 poolformer 1,326
23 involution 1,307

Sponsored
InfluxDB – Built for High-Performance Time Series Workloads
InfluxDB 3 OSS is now GA. Transform, enrich, and act on time series data directly in the database. Automate critical tasks and eliminate the need to move data externally. Download now.
www.influxdata.com

Did you know that Python is
the 2nd most popular programming language
based on number of references?