Python image-classification

Open-source Python projects categorized as image-classification

Top 23 Python image-classification Projects

  • 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.

    Project mention: labelImg: NEW Data - star count:19756.0 | | 2023-05-13
  • vit-pytorch

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

    Project mention: Is it easier to go from Pytorch to TF and Keras than the other way around? | | 2023-05-13

    I also need to learn Pyspark so right now I am going to download the Fashion Mnist dataset, use Pyspark to downsize each image and put the into separate folders according to their labels (just to show employers I can do some basic ETL with Pyspark, not sure how I am going to load for training in Pytorch yet though). Then I am going to write the simplest Le Net to try to categorize the fashion MNIST dataset (results will most likely be bad but it's okay). Next, try to learn transfer learning in Pytorch for both CNN or maybe skip ahead to ViT. Ideally at this point I want to study the Attention mechanism a bit more and try to implement Simple Vit which I saw here:

  • InfluxDB

    Access the most powerful time series database as a service. Ingest, store, & analyze all types of time series data in a fully-managed, purpose-built database. Keep data forever with low-cost storage and superior data compression.

  • label-studio

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

    Project mention: Best (quickest) way to annotate images for whole-image classification? | | 2023-05-21

    LabelStudio is free for single use.

  • albumentations

    Fast image augmentation library and an easy-to-use wrapper around other libraries. Documentation: Paper about the library:

    Project mention: Burn Deep Learning Framework Release 0.7.0: Revamped (de)serialization, optimizer & module overhaul, initial ONNX support and tons of new features. | | 2023-05-06

    Is something planned to support data augmentations? Something like

  • Swin-Transformer

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

    Project mention: Swin Transformer: Hierarchical Vision Transformer using Shifted Windows | | 2023-04-09
  • pytorch-grad-cam

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

    Project mention: Which layers are doing image segmentation on AutoEncoders/U-NET? | | 2022-07-10

  • cleanlab

    The standard data-centric AI package for data quality and machine learning with messy, real-world data and labels.

    Project mention: [R] Automated Checks for Violations of Independent and Identically Distributed (IID) Assumption | | 2023-05-30

    I just published a paper detailing this non-IID check and open-sourced its code in the cleanlab package — just one line of code will check for this and many other types of issues in your dataset.

  • CodiumAI

    TestGPT | Generating meaningful tests for busy devs. Get non-trivial tests (and trivial, too!) suggested right inside your IDE, so you can code smart, create more value, and stay confident when you push.

  • autogluon

    AutoGluon: AutoML for Image, Text, Time Series, and Tabular Data

    Project mention: pip install remyxai - easiest way to create custom vision models | | 2023-04-25

    This seems not very convincing. There are other popular frameworks that provide AutoML with existing datasets (eg

  • gluon-cv

    Gluon CV Toolkit

    Project mention: FOSS self-hosted image-to-text gpu accelerated object recognition ? Is there anything on the table yet ? | | 2023-03-30

  • PaddleClas

    A treasure chest for visual classification and recognition powered by PaddlePaddle

  • hub

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

  • mmpretrain

    OpenMMLab Pre-training Toolbox and Benchmark

    Project mention: MMDeploy: Deploy All the Algorithms of OpenMMLab | | 2022-11-21

    MMClassification: OpenMMLab image classification toolbox and benchmark.

  • efficientnet

    Implementation of EfficientNet model. Keras and TensorFlow Keras.

    Project mention: Why did the original ResNet paper not use dropout? | | 2023-01-29

    not true at all, plenty of sota models combines batchnorm and dropout 1. efficientnet 2. resnet rs 3. timm resnet50 (appendix)

  • sparseml

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

    Project mention: Can You Achieve GPU Performance When Running CNNs on a CPU? | | 2023-04-08
  • ailia-models

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

    Project mention: 10$ Full Body Tracking! I'm proud to release ToucanTrack (in Beta!). Get decent FBT with the power of 2 PS3 Eye Cameras and AI! | | 2023-02-20

    If you're looking for the differences in terms of how inference is done, I recommend you take a look at MediaPipe's source code. MediaPipe doesn't use raw code, but uses a "graph" instead (eg. pose_landmark_cpu.pbtxt), which can be visualised using MediaPipe Viz. I also used axinc-ai/ailia-models as the base (preprocessing, inference, postprocessing, etc...) which I further built upon (keypoint refinement, roi from keypoints, filtering / smoothing, etc...)

  • pytorch-toolbelt

    PyTorch extensions for fast R&D prototyping and Kaggle farming

    Project mention: Surrender cards are being distributed to Russianforces: 'Your ticket to a peaceful life. Show this card to a Ukrainian soldier - it will save your life and help you get back home' On the back: a telegram chat & phone number they can contact 'to receive detailed support'. | | 2022-09-12

    For what it's worth, the QR code points to a URI shortener domain "" which redirects to a cloudflare hosted page that contains telegram information relating to "nikolay_bodenko" and "chaos_admin" ("For advertising and cooperation"? Whatever that is supposed to mean), which led me to "rf200_now" and "rf200_nooow" (looks to get spammed around a bit), which is purportedly supportive of Ukraine.

  • involution

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

  • bootcamp

    Dealing with all unstructured data, such as reverse image search, audio search, molecular search, video analysis, question and answer systems, NLP, etc. (by milvus-io)

    Project mention: Beginner-ish resources for choosing a vector database? | | 2023-05-20

    Easy to get started: Here are some tutorials for Milvus in a Jupyter Notebook that I wrote - reverse image search, semantic text search

  • Unsupervised-Classification

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

    Project mention: Middle ground dataset between CIFAR and ImageNet [D] | | 2022-09-17

    The subsets we used are from here:

  • private-detector

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

    Project mention: Hotdog. Not hotdog. | | 2022-11-23

    Remember hotdog/not hotdog? yeah, somebody has released an image classifier (link to source on GitHub) that does just that (for body parts, not hotdogs).

  • SimMIM

    This is an official implementation for "SimMIM: A Simple Framework for Masked Image Modeling".

    Project mention: My DL model mis-segment wrong objects with similar colors | | 2023-03-20

    It seems the segmentation task is complex. One thing that I can suggest that definitely worked for me is Self-supervised learning prior to training. The task of the sel-supervision is very important. In your case I suggest Masked Image Modeling because it does not rely heavily on shapes and colors but can leverage them if they exist. Check out SimMIM by Microsoft. .

  • caer

    High-performance Vision library in Python. Scale your research, not boilerplate.

  • UniFormer

    [ICLR2022] official implementation of UniFormer


    ONLYOFFICE Docs — document collaboration in your environment. Powerful document editing and collaboration in your app or environment. Ultimate security, API and 30+ ready connectors, SaaS or on-premises

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). The latest post mention was on 2023-05-30.

Python image-classification related posts


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

Project Stars
1 labelImg 19,916
2 vit-pytorch 14,194
3 label-studio 13,046
4 albumentations 12,044
5 Swin-Transformer 10,888
6 pytorch-grad-cam 7,464
7 cleanlab 5,935
8 autogluon 5,771
9 gluon-cv 5,551
10 PaddleClas 4,847
11 hub 3,335
12 mmpretrain 2,239
13 efficientnet 2,025
14 sparseml 1,652
15 ailia-models 1,414
16 pytorch-toolbelt 1,394
17 involution 1,279
18 bootcamp 1,211
19 Unsupervised-Classification 1,186
20 private-detector 1,134
21 SimMIM 733
22 caer 688
23 UniFormer 637
Write Clean Python Code. Always.
Sonar helps you commit clean code every time. With over 225 unique rules to find Python bugs, code smells & vulnerabilities, Sonar finds the issues while you focus on the work.