Top 23 Python image-classification Projects

image-classification

• Add a project

1. ultralytics

   1 36 40,251 9.9 Python

   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

   www.influxdata.com featured

   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.

   

3. pytorch-image-models

   2 37 34,064 9.6 Python

   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

   3 11 22,795 7.6 Python

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

   

5. albumentations

   4 38 14,908 9.6 Python

   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

   5 23 14,746 2.9 Python

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

   

7. pytorch-grad-cam

   6 5 11,612 7.0 Python

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

   

8. fiftyone

   7 32 9,461 10.0 Python

   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

   www.saashub.com featured

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

   

10. InternVL

    8 1 8,015 8.9 Python

    [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

    9 1 5,882 0.0 Python

    Gluon CV Toolkit

    

12. PaddleClas

    10 2 5,644 7.9 Python

    A treasure chest for visual classification and recognition powered by PaddlePaddle

    

13. mmpretrain

    11 2 3,649 2.9 Python

    OpenMMLab Pre-training Toolbox and Benchmark

    

14. hub

    12 1 3,496 1.3 Python

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

    

15. catalyst

    13 1 3,336 0.0 Python

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

    

16. autodistill

    14 20 2,252 7.0 Python

    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

    15 4 2,191 9.7 Python

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

    

18. sparseml

    16 12 2,131 5.3 Python

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

    

19. efficientnet

    17 9 2,083 0.0 Python

    Implementation of EfficientNet model. Keras and TensorFlow Keras.

    

20. fastdup

    18 19 1,680 8.7 Python

    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

    19 1 1,541 7.8 Python

    PyTorch extensions for fast R&D prototyping and Kaggle farming

    

22. Unsupervised-Classification

    20 2 1,418 1.4 Python

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

    

23. private-detector

    21 5 1,330 4.6 Python

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

    

24. poolformer

    22 3 1,326 3.0 Python

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

    

25. involution

    23 6 1,307 0.0 Python

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

    

26. SaaSHub

    www.saashub.com featured

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

    

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

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

Do not miss the trending Python projects with our weekly report!