Repository Index

Ultralytics YOLO27, YOLO26, YOLO11, YOLOv8 — object detection, instance segmentation, semantic segmentation, image classification, pose estimation, object tracking

• ultralytics
• yolov8
• yolo
• deep-learning

Ultralytics YOLOv5 in PyTorch for object detection, instance segmentation, classification, training, and export.

• yolov5
• object-detection
• pytorch
• onnx

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

• pytorch
• resnet
• pretrained-models
• pretrained-weights

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

• computer-vision
• deep-learning
• image-annotation
• annotation-tool

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

• artificial-intelligence
• attention-mechanism
• transformers
• computer-vision

A Lightweight Face Recognition and Facial Attribute Analysis (Age, Gender, Emotion and Race) Library for Python

• face-recognition
• vgg-face
• facenet
• openface

CVAT is a leading data annotation platform for image, video, audio, and 3D datasets. It offers open-source, cloud, and enterprise products, as well as labeling services, with AI-assisted labeling, quality assurance, team collaboration, analytics, and developer APIs.

• video-annotation
• computer-vision
• computer-vision-annotation
• image-annotation

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

• deep-learning
• pytorch
• grad-cam
• visualizations

Refine high-quality datasets and visual AI models

• machine-learning
• artificial-intelligence
• deep-learning
• computer-vision

X-AnyLabeling: A lightweight, efficient, and unified cross-platform desktop application for annotating text, image, video, point cloud and multimodal data, combining versatile built-in tools with state-of-the-art AI models and flexible multi-format export.

• sam
• yolo
• deep-learning
• object-detection

Translated[CVPR 2024 Oral] InternVL Family: A Pioneering Open-Source Alternative to GPT-4o. An open-source multimodal dialogue model with performance close to GPT-4o

• image-classification
• image-text-retrieval
• llm
• semantic-segmentation

all kinds of text classification models and more with deep learning

• classification
• nlp
• fasttext
• textcnn

Accelerated deep learning R&D

• deep-learning
• reinforcement-learning
• machine-learning
• computer-vision

ThunderSVM: A Fast SVM Library on GPUs and CPUs

• classification
• regression
• one-class-learning
• cuda

Official PyTorch implementation of VisionHOPE: Visual Backbones as Self-Modifying Learning Systems.

• backbone
• computer-vision
• image-classification
• backbone-networks

The most accurate natural language detection library for Go, suitable for short text and mixed-language text

• natural-language-processing
• language-detection
• language-recognition
• language-classification

TorchXRayVision: A library of chest X-ray datasets and models. Classifiers, segmentation, and autoencoders.

• medical-imaging
• medical-ai
• deep-learning
• machine-learning

Classify music genre from a 10 second sound stream using a Neural Network.

• deep-learning
• music-genre-classification
• paper
• tensorflow

AgML is a centralized framework for agricultural machine learning. AgML provides access to public agricultural datasets for common agricultural deep learning tasks, with standard benchmarks and pretrained models, as well the ability to generate synthetic data and annotations.

• deep-learning
• agriculture
• pytorch
• dataset

:wink: :cyclone: :strawberry: TextRank implementation in Golang with extendable features (summarization, phrase extraction) and multithreading (goroutine).

• go
• golang
• textrank
• summarization

A simple, flexible, and extensible template for PyTorch. It's beautiful.

• deep-learning
• computer-vision
• cnn
• pytorch

JevStyle: local decision models for AI agents. Classify, route and score with a probability per option. systemone-compatible local server, MCP tools, Claude Code guard. Open weights, Apache-2.0.

• agent-skills
• calibration
• claude-code
• decision-model

A curated dataset of malware and benign Windows executable samples for malware researchers

• machine-learning
• deep-learning
• malware-research
• malware-detection

Golem Image Classifier is built using the Service API of Yapapi to interact with the Golem Network (golem.network). Built as part of a Gitcoin Bounty

• machine-learning
• golem
• yagna