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. Learn more →
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.
-
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? | reddit.com/r/pytorch | 2023-05-13I 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: https://github.com/lucidrains/vit-pytorch/blob/main/vit_pytorch/simple_vit.py
-
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? | reddit.com/r/learnmachinelearning | 2023-05-21LabelStudio is free for single use. https://labelstud.io/
-
albumentations
Fast image augmentation library and an easy-to-use wrapper around other libraries. Documentation: https://albumentations.ai/docs/ Paper about the library: https://www.mdpi.com/2078-2489/11/2/125
Project mention: Burn Deep Learning Framework Release 0.7.0: Revamped (de)serialization, optimizer & module overhaul, initial ONNX support and tons of new features. | reddit.com/r/rust | 2023-05-06Is something planned to support data augmentations? Something like https://albumentations.ai/
-
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 | reddit.com/r/neuralnetworks | 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? | reddit.com/r/computervision | 2022-07-10https://github.com/jacobgil/pytorch-grad-cam.
-
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 | reddit.com/r/MachineLearning | 2023-05-30I 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.
-
Project mention: pip install remyxai - easiest way to create custom vision models | reddit.com/r/computervision | 2023-04-25
This seems not very convincing. There are other popular frameworks that provide AutoML with existing datasets (eg https://github.com/autogluon/autogluon)
-
Project mention: FOSS self-hosted image-to-text gpu accelerated object recognition ? Is there anything on the table yet ? | reddit.com/r/selfhosted | 2023-03-30
https://github.com/amusi/awesome-object-detection https://mmdetection.readthedocs.io/en/latest/index.html https://github.com/thtrieu/darkflow https://github.com/OlafenwaMoses/ImageAI https://github.com/dmlc/gluon-cv https://github.com/aim-uofa/AdelaiDet/ https://github.com/aim-uofa/AdelaiDet/blob/master/configs/FCOS-Detection/README.md https://github.com/wizyoung/YOLOv3_TensorFlow
-
-
-
Project mention: MMDeploy: Deploy All the Algorithms of OpenMMLab | reddit.com/r/u_Allent_pjlab | 2022-11-21
MMClassification: OpenMMLab image classification toolbox and benchmark.
-
Project mention: Why did the original ResNet paper not use dropout? | reddit.com/r/datascience | 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? | reddit.com/r/computervision | 2023-04-08 -
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! | reddit.com/r/VRchat | 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...)
-
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'. | reddit.com/r/ukraine | 2022-09-12
For what it's worth, the QR code points to a URI shortener domain "qrfy.mobi" 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? | reddit.com/r/vectordatabase | 2023-05-20Easy 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] | reddit.com/r/MachineLearning | 2022-09-17The subsets we used are from here: https://github.com/wvangansbeke/Unsupervised-Classification/tree/master/data/imagenet_subsets
-
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 | reddit.com/r/computervision | 2023-03-20It 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. https://arxiv.org/abs/2111.09886 . https://github.com/microsoft/simmim
-
-
-
ONLYOFFICE
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
Python image-classification related posts
- Is it easier to go from Pytorch to TF and Keras than the other way around?
- Swin Transformer: Hierarchical Vision Transformer Using Shifted Windows
- Swin Transformer: Hierarchical Vision Transformer using Shifted Windows
- Swin Transformer: Hierarchical Vision Transformer Using Shifted Windows
- PyTorch at the Edge: Deploy 964 TIMM Models on Android with TorchScript
- Why did the original ResNet paper not use dropout?
- [D] Influential papers round-up 2022. What are your favorites?
-
A note from our sponsor - Sonar
www.sonarsource.com | 30 May 2023
Index
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 |