Python image-classification

Open-source Python projects categorized as image-classification | Edit details

Top 23 Python image-classification Projects

  • GitHub repo labelImg

    🖍️ LabelImg is a graphical image annotation tool and label object bounding boxes in images

    Project mention: Anyone know of a free tool to help with preparing the JSON file for CreateMl Object Detection? | | 2021-12-12
  • GitHub repo albumentations

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

    Project mention: Data augmentation strategies for object detection? Could you point me to good resources or best practices you know of? | | 2021-10-30

    You can definitely look at Albumentation - we had a ton of success working with this library

  • Scout APM

    Less time debugging, more time building. Scout APM allows you to find and fix performance issues with no hassle. Now with error monitoring and external services monitoring, Scout is a developer's best friend when it comes to application development.

  • GitHub repo 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: [D] Surprisingly Simple SOTA Self-Supervised Pretraining - Masked Autoencoders Are Scalable Vision Learners by Kaiming He et al. explained (5-minute summary by Casual GAN Papers) | | 2021-11-17

    nah, it is really simple. here is the code

  • GitHub repo label-studio

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

    Project mention: [D] Portals for outsourcing preliminary data labeling | | 2022-01-13

    Not exactly for this solution, but I have really liked this tool. It is open source and can be self hosted if needed

  • GitHub repo autogluon

    AutoGluon: AutoML for Text, Image, and Tabular Data

    Project mention: What will the data science job market be like in 5 years? | | 2021-08-14

    Some AutoML is getting pretty good, AutoGluon is very solid for tabular data. That being said you still need to have your data in tabular format and deployment still requires some effort.

  • GitHub repo PaddleClas

    A treasure chest for visual recognition powered by PaddlePaddle

    Project mention: Baidu AI Research Team Introduces ‘PP-ShiTu’: A Practical Lightweight Image Recognition System | | 2021-11-07

    Quick 5 Min Read | Paper | Github

  • GitHub repo efficientnet

    Implementation of EfficientNet model. Keras and TensorFlow Keras.

    Project mention: I made an image recognition model written in NodeJs | | 2021-02-24

    EfficientNet a lightweight convolutional neural network architecture achieving the state-of-the-art accuracy with an order of magnitude fewer parameters and FLOPS, on both ImageNet and five other commonly used transfer learning datasets.

  • OPS

    OPS - Build and Run Open Source Unikernels. Quickly and easily build and deploy open source unikernels in tens of seconds. Deploy in any language to any cloud.

  • GitHub repo involution

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

    Project mention: [R] Involution: Inverting the Inherence of Convolution for Visual Recognition | | 2021-05-10

    PDF Link | Landing Page | Read as web page on arXiv Vanity

  • GitHub repo mmclassification

    OpenMMLab Image Classification Toolbox and Benchmark

    Project mention: how to recognize digits from this pics(i have many of them) so kindly suggest generic that can work for other similar images. I have searched alot for the source code on github but not found the correct solution. most of these solutions were incorrect while other were incomplete. Kindly help me :( | | 2021-02-09

    MMClassification or TIMM would be good starting points for training a classification model.

  • GitHub repo Unsupervised-Classification

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

    Project mention: Any reference or idea about how to train unsupervised CNN model ? | | 2021-04-13
  • GitHub repo assembled-cnn

    Tensorflow implementation of "Compounding the Performance Improvements of Assembled Techniques in a Convolutional Neural Network"

    Project mention: [R] ResNet strikes back: An improved training procedure in timm. There has been significant progress on best practices for training neural nets since ResNet's introduction in 2015. With such advances, a vanilla ResNet-50 reaches 80.4% top-1 accuracy on ImageNet without extra data or distillation. | | 2021-10-04

    As far as i know, the assemble-ResNet-50 ( gets 82.8% top-1, though they make some (minor) changes to ResNet-50 architecture.

  • GitHub repo transformer-in-transformer

    Implementation of Transformer in Transformer, pixel level attention paired with patch level attention for image classification, in Pytorch

    Project mention: “Transformer in Transformer” paper explained! | | 2021-03-04

    A thirdparty implementation of " Transformer in Transformer":

  • GitHub repo groupImg

    A script in python to organize your images by similarity.

  • GitHub repo chitra

    A multi-functional library for full-stack Deep Learning. Simplifies Model Building, API development, and Model Deployment.

    Project mention: Answer: Resizing image and its bounding box | | 2021-07-03

    Another way of doing this is to use CHITRA

  • GitHub repo convolution-vision-transformers

    PyTorch Implementation of CvT: Introducing Convolutions to Vision Transformers

    Project mention: CvT: Introducing Convolutions to Vision Transformers | | 2021-03-30


  • GitHub repo MLP-Mixer-pytorch

    Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Vision

    Project mention: [R] MLP-Mixer: An all-MLP Architecture for Vision | | 2021-05-05

    My implementation:

  • GitHub repo labelme2coco

    A lightweight package for converting your labelme annotations into COCO object detection format.

    Project mention: What's A Simple Custom Segmentation Pipeline? | | 2021-02-12

    I would also suggest labelme, it's pretty easy to use. Just type "labelme" in the shell after pip installing and you will see the GUI. There are tools to convert to coco format (like if needed, for instance for Detectron2.

  • GitHub repo image-sorter2

    One-click image sorting/labelling script

    Project mention: How are custom (bounding box) object datasets collected in research/practice? Thinking about making an iOS app to help if this is tedious. | | 2021-05-23

    I first altered the image-sorter2 code to do multi-classes and save to CSV file instead of moving files to directories. I then implemented the YOLOv5 algorithm in the image viewing, so that it would predict where the people were in the image. I could then label where the people were by box number and label their activity. It ends up being pretty quick to label images this way.

  • GitHub repo CeiT

    Implementation of Convolutional enhanced image Transformer

    Project mention: [2103.11816] Incorporating Convolution Designs into Visual Transformers | | 2021-03-26


  • GitHub repo subwAI

    Scripts for training an AI to play the endless runner Subway Surfers using a supervised machine learning approach by imitation and a convolutional neural network (CNN) for image classification

    Project mention: Started programming with cs50x last and ai50 this year - I expanded one of the psets to make an ai play a game | | 2021-09-24
  • GitHub repo targetran

    Python library for data augmentation in object detection or image classification model training

    Project mention: New image augmentation library for TF Dataset + TPU | | 2021-09-14
  • GitHub repo pytorch_deep_learning

    Using Pytorch to Create Deep Learning Models.

    Project mention: Using PyTorch to Create Deep Learning Models. | | 2021-12-08

    I've created a repository ‏‏‎ full of PyTorch codes for different deep learning models, please finish check out and any feedback is appreciated.

  • GitHub repo autoCustomVision

    GitHub Action automating the entire process of image classification with Microsoft Azure Custom Vision Service

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 2022-01-13.

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 16,305
2 albumentations 9,493
3 vit-pytorch 8,219
4 label-studio 7,380
5 autogluon 4,042
6 PaddleClas 3,230
7 efficientnet 1,929
8 involution 1,212
9 mmclassification 1,024
10 Unsupervised-Classification 896
11 assembled-cnn 328
12 transformer-in-transformer 257
13 groupImg 181
14 chitra 170
15 convolution-vision-transformers 160
16 MLP-Mixer-pytorch 158
17 labelme2coco 76
18 image-sorter2 60
19 CeiT 59
20 subwAI 57
21 targetran 10
22 pytorch_deep_learning 7
23 autoCustomVision 4
Find remote jobs at our new job board There are 29 new remote jobs listed recently.
Are you hiring? Post a new remote job listing for free.
Deliver Cleaner and Safer Code - Right in Your IDE of Choice!
SonarLint is a free and open source IDE extension that identifies and catches bugs and vulnerabilities as you code, directly in the IDE. Install from your favorite IDE marketplace today.