Topics / #image-classification

Image-classification projects on GitHub

Vision models that assign labels to images.

As of , the most-starred Image-classification projects on GitHub are ultralytics (62.6K stars), yolov5 (58.2K stars) and pytorch-image-models (37.2K stars). GitHub.Locker tracks 47 Image-classification repositories with 865.3K stars combined. The fastest-growing on October 16 was VisionHOPE, with +111 stars in a day.

47Projects
865.3KStars in total
+281Stars in a day
Browse in the catalog →github.com/topics/image-classification ↗

Hot right now

Most stars gained in a day · Oct 16

+111

VisionHOPE

PSRben/VisionHOPE · ★ 1.2K

VisionHOPE is the official PyTorch implementation of a visual backbone built on self-referential nested learning (SRNL), supporting ImageNet-1K classification, COCO detection/instance segmentation, and ADE20K semantic segmentation.

+57

autoshade

skymanbp/autoshade · ★ 1.1K

AutoShade is an AI-assisted RAW photo developer: you describe the look you want, a GPT-style vision step proposes an edit recipe, and a deterministic Rust engine renders that recipe onto the full-resolution RAW file — producing Lightroom-compatible XMP sidecars instead of AI-generated pixels.

+29

ultralytics

ultralytics/ultralytics · ★ 62.6K

Ultralytics YOLO is a family of state-of-the-art computer vision models (YOLO26, YOLO11, YOLOv8, etc.) covering object detection, instance segmentation, semantic segmentation, image classification, pose estimation, oriented object detection, depth estimation, and multi-object tracking in video.

+11

X-AnyLabeling

CVHub520/X-AnyLabeling · ★ 10.7K

A free cross-platform desktop app for AI-assisted annotation of text, image, video, and multimodal data — manual labeling tools plus built-in deep learning models that pre-label or batch-annotate, with multi-format export.

The all-time top

47 projects · by stars

  1. #1 · ultralytics/ultralytics

    ultralytics

    Ultralytics YOLO is a family of state-of-the-art computer vision models (YOLO26, YOLO11, YOLOv8, etc.) covering object detection, instance segmentation, semantic segmentation, image classification, pose estimation, oriented object detection, depth estimation, and multi-object tracking in video.

    62.6K
  2. #2 · ultralytics/yolov5

    yolov5

    YOLOv5 is a fast, accurate, and easy-to-use computer vision model from Ultralytics for object detection, instance segmentation, and image classification, built on PyTorch.

    58.2K
  3. #3 · huggingface/pytorch-image-models

    pytorch-image-models

    PyTorch Image Models (timm) — the largest collection of PyTorch image encoders and backbones, shipped with pretrained weights plus train, eval, inference and export scripts.

    37.2K
  1. 4

    detectron2

    34.9Kfacebookresearch/detectron2 · Python

    Detectron2 is Facebook AI Research's next-generation library for object detection, segmentation and other visual recognition tasks, with state-of-the-art algorithms.

  2. 5

    CLIP

    34.5Kopenai/CLIP · Jupyter Notebook

    CLIP is a neural network trained on many (image, text) pairs: given an image, it predicts the most relevant text snippet in natural language, zero-shot, without directly optimizing for the task.

  3. 6

    EasyOCR

    30.1KJaidedAI/EasyOCR · Python

    A ready-to-use, open-source OCR library for Python: it detects and reads text in images across 80+ languages and popular writing scripts — Latin, Chinese, Arabic, Devanagari, Cyrillic and more.

  4. 7

    label-studio

    28.5KHumanSignal/label-studio · TypeScript

    Label Studio is an open source data labeling and annotation tool: you label audio, text, images, videos, and time series in a browser UI and export the results in formats ready for ML training.

  5. 8

    deep-learning-for-image-processing

    26.4KWZMIAOMIAO/deep-learning-for-image-processing · Python

    A video-based deep learning course that walks through classic computer-vision architectures for image classification, object detection, segmentation and keypoint detection, pairing each network with a lecture and runnable code.

  6. 9

    Detectron

    26.4Kfacebookresearch/Detectron · Python

    Detectron is Facebook AI Research's software system implementing state-of-the-art object detection algorithms like Mask R-CNN and RetinaNet, built as a research platform for computer vision.

  7. 10

    vit-pytorch

    25.6Klucidrains/vit-pytorch · Python

    A PyTorch implementation of the Vision Transformer (ViT), an image classification model that reaches SOTA results using just a single transformer encoder instead of convolutions.

  8. 11

    fastbook

    25.4Kfastai/fastbook · Jupyter Notebook

    This repo contains the official Jupyter Notebooks for 'the fastai book', an introduction to deep learning, the fastai library, and PyTorch.

  9. 12

    labelImg

    25KHumanSignal/labelImg · Python

    A graphical image annotation tool: you draw bounding boxes around objects in images to build labeled datasets for object detection. Created by Tzutalin, it now lives with the Label Studio community.

  10. 13

    pytorch-handbook

    21.7Kzergtant/pytorch-handbook · Jupyter Notebook

    An open-source Chinese-language book that teaches PyTorch for deep learning, written as a set of Jupyter notebooks whose code has all been tested to actually run.

  11. 14

    vision

    18Kpytorch/vision · Python

    Torchvision is PyTorch's computer vision package: popular datasets, ready-made model architectures, and common image transformations for vision work.

  12. 15

    DeepLearning

    17.8KMikoto10032/DeepLearning · Jupyter Notebook

    A Chinese-language deep learning tutorial — a huge curated collection of links to courses, books, lecture notes, surveys and code, taking you from math basics through CNNs, GANs, RNNs and GNNs to Kaggle practice and interview prep.

  13. 16

    tensor2tensor

    17.5Ktensorflow/tensor2tensor · Python

    A library of deep learning models and datasets from researchers and engineers on the Google Brain team, built to make deep learning more accessible and speed up ML research.

  14. 17

    cvat

    16.9Kcvat-ai/cvat · Python

    CVAT is an open-source, self-hosted data annotation platform for building labeled datasets from images, video, and 3D point clouds, used for computer vision and visual AI projects.

  15. 18

    transformers.js

    16.3Khuggingface/transformers.js · JavaScript

    Transformers.js lets you run Hugging Face's state-of-the-art machine learning models directly in a web browser, with no server required, using an API designed to mirror the Python transformers library.

  16. 19

    Swin-Transformer

    16.1Kmicrosoft/Swin-Transformer · Python

    The official Microsoft implementation of the Swin Transformer paper and its follow-ups: a hierarchical vision Transformer that computes attention with shifted windows and serves as a general-purpose backbone for computer vision tasks.

  17. 20

    openface

    15.4Kcmusatyalab/openface · Lua

    OpenFace is a free, open-source face recognition toolkit built on deep neural networks, developed by Carnegie Mellon University.

  18. 21

    albumentations

    15.3Kalbumentations-team/albumentations · Python

    Albumentations is a fast, flexible Python library for image augmentation used in training computer vision models, offering a huge set of pixel-level and spatial-level transforms for images, masks, bounding boxes, keypoints, and even 3D volumes.

  19. 22

    DeepLearningExamples

    14.9KNVIDIA/DeepLearningExamples · Jupyter Notebook

    A collection of NVIDIA-maintained, state-of-the-art deep learning reference implementations across computer vision, NLP, recommenders, speech, and more, tuned for reproducible accuracy and performance on NVIDIA Tensor Core GPUs.

  20. 23

    tfjs-models

    14.8Ktensorflow/tfjs-models · TypeScript

    A collection of pretrained machine learning models ported to TensorFlow.js, covering vision, audio, text, and general utility tasks, ready to use directly in browser or Node.js projects.

  21. 24

    imgaug

    14.7Kaleju/imgaug · Python

    imgaug is a Python library that augments images for machine learning by converting a set of input images into a much larger set of slightly altered versions.

  22. 25

    facenet

    14.4Kdavidsandberg/facenet · Python

    This is a TensorFlow implementation of FaceNet, a face recognition system that converts face images into 128-dimensional embeddings for recognition and clustering, based on the paper 'FaceNet: A Unified Embedding for Face Recognition and Clustering'.

  23. 26

    dinov2

    13.4Kfacebookresearch/dinov2 · Jupyter Notebook

    DINOv2, a self-supervised visual feature learning method from Meta AI's FAIR team, shipped as PyTorch code and pretrained models. Its features plug into classifiers as simple as linear layers across many vision tasks, with no fine-tuning required.

  24. 27

    pytorch-grad-cam

    13Kjacobgil/pytorch-grad-cam · Python

    A Python package that implements many state-of-the-art Explainable AI (pixel attribution/Class Activation Map) methods for computer vision models, letting you visualize which parts of an image a model focuses on.

  25. 28

    fashion-mnist

    12.9Kzalandoresearch/fashion-mnist · Python

    A dataset of Zalando's article images for benchmarking machine learning algorithms: 60,000 training and 10,000 test examples, each a 28x28 grayscale image with a label from 10 classes. It is meant as a direct drop-in replacement for the original MNIST.

  26. 29

    pytorch-book

    12.8Kchenyuntc/pytorch-book · Jupyter Notebook

    This is the companion code repository for the Chinese book 'Deep Learning Framework PyTorch: Introduction and Practice', covering PyTorch basics and fun projects like neural talk, poem writing, anime face generation, and style transfer.

  27. 30

    vision_transformer

    12.7Kgoogle-research/vision_transformer · Jupyter Notebook

    This is Google Research's official JAX/Flax code release for Vision Transformer (ViT) and MLP-Mixer image classification models, including pretrained checkpoints and scripts to fine-tune them on your own data.

  28. 31

    nsfw_data_scraper

    12.6Kalex000kim/nsfw_data_scraper · Shell

    This is a collection of shell scripts that scrapes and aggregates tens of thousands of images across 5 categories (porn, hentai, sexy, neutral, drawings) to build a labeled dataset for training an NSFW image classifier.

  29. 32

    CVPR2024-Paper-Code-Interpretation

    12.5Kextreme-assistant/CVPR2024-Paper-Code-Interpretation

    A curated archive of CVPR conference papers, code links, Chinese-language write-ups, and livestream talk collections spanning CVPR2017 through CVPR2024, compiled by the Extreme Mart (极市) team.

  30. 33

    External-Attention-pytorch

    12.2Kxmu-xiaoma666/External-Attention-pytorch · Python

    A PyTorch collection of Attention, Backbone, MLP, Re-parameter, and Convolution implementations from research papers, meant to help you understand papers by seeing their mechanisms coded up.

  31. 34

    lime

    12.2Kmarcotcr/lime · JavaScript

    LIME (Local Interpretable Model-agnostic Explanations) is a Python package that explains individual predictions of any machine learning classifier, whether it works on text, tabular data, or images.

  32. 35

    Transformers-Tutorials

    11.8KNielsRogge/Transformers-Tutorials · Jupyter Notebook

    A collection of Jupyter notebook demos showing how to use HuggingFace's Transformers library in PyTorch, covering models like BERT, GPT-2, LayoutLM, DETR, Donut and many more.

  33. 36

    dinov3

    11.6Kfacebookresearch/dinov3 · Jupyter Notebook

    Meta AI Research (FAIR)'s reference PyTorch implementation and model release for DINOv3, a family of vision foundation models that produce high-quality dense features and perform strongly across many vision tasks.

  34. 37

    fiftyone

    11.2Kvoxel51/fiftyone · TypeScript

    FiftyOne is an open-source tool for building high-quality datasets and computer vision models: you use it to visualize and label your visual data, evaluate models, and find and fix data quality issues.

  35. 38

    awesome-deep-vision

    11.2Kkjw0612/awesome-deep-vision

    A curated list of deep learning resources for computer vision, covering papers on classification, detection, segmentation and more, plus courses, books, software and tutorials.

  36. 39

    turicreate

    11.2Kapple/turicreate · C++

    Turi Create is a Python library from Apple that lets you build custom machine learning models for tasks like recommendations, object detection, image classification, image similarity, and activity classification without being an ML expert.

  37. 40

    cs231n.github.io

    11.1Kcs231n/cs231n.github.io · Jupyter Notebook

    This is the public notes and assignments page for Stanford's CS231n course on Convolutional Neural Networks for Visual Recognition.

  38. 41

    awesome-semantic-segmentation

    10.8Kmrgloom/awesome-semantic-segmentation

    A curated list of resources for semantic segmentation: network architectures, papers, code implementations, datasets, benchmarks, annotation tools, and losses used in the field.

  39. 42

    X-AnyLabeling

    10.7KCVHub520/X-AnyLabeling · Python

    A free cross-platform desktop app for AI-assisted annotation of text, image, video, and multimodal data — manual labeling tools plus built-in deep learning models that pre-label or batch-annotate, with multi-format export.

  40. 43

    techniques

    10.3Ksatellite-image-deep-learning/techniques

    A curated list of techniques for deep learning with satellite and aerial imagery, covering architectures, models and algorithms for tasks like classification, segmentation and object detection.

  41. 44

    opencv_contrib

    10.2Kopencv/opencv_contrib · C++

    This is the repository for OpenCV's extra modules — contributed computer vision functionality that isn't stable or tested enough for the core OpenCV distribution yet.

  42. 45

    InternVL

    10.2KOpenGVLab/InternVL · Python

    A family of open-source multimodal chat models that understand images, video and text, positioned as an open alternative to GPT-4o. InternVL2.5-78B was the first open-source model to top 70% on the MMMU benchmark.

  43. 46

    VisionHOPE

    1.2KPSRben/VisionHOPE · Python

    VisionHOPE is the official PyTorch implementation of a visual backbone built on self-referential nested learning (SRNL), supporting ImageNet-1K classification, COCO detection/instance segmentation, and ADE20K semantic segmentation.

  44. 47

    autoshade

    1.1Kskymanbp/autoshade · Rust

    AutoShade is an AI-assisted RAW photo developer: you describe the look you want, a GPT-style vision step proposes an edit recipe, and a deterministic Rust engine renders that recipe onto the full-resolution RAW file — producing Lightroom-compatible XMP sidecars instead of AI-generated pixels.

Often tagged together