Repo for counting stars and contributing. Press F to pay respect to glorious developers.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

The repository provides code for running inference with the SegmentAnything Model (SAM), links for downloading the trained model checkpoints, and example notebooks that show how to use the model.

Ultralytics YOLO 🚀

TensorFlow code and pre-trained models for BERT

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.

CLIP (Contrastive Language-Image Pretraining), Predict the most relevant text snippet given an image

Convolutional Neural Networks

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 …

[NeurIPS'23 Oral] Visual Instruction Tuning (LLaVA) built towards GPT-4V level capabilities and beyond.

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

YOLOv4 / Scaled-YOLOv4 / YOLO - Neural Networks for Object Detection (Windows and Linux version of Darknet )

Original reference implementation of "3D Gaussian Splatting for Real-Time Radiance Field Rendering"

Mamba SSM architecture

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

End-to-End Object Detection with Transformers

Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors

YOLOv3 in PyTorch > ONNX > CoreML > TFLite

YOLOX is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported. Documentation: https://yolox.readthedocs.io/

Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.

PyTorch implementation of MAE https//arxiv.org/abs/2111.06377

Object detection, 3D detection, and pose estimation using center point detection:

Awesome Object Detection based on handong1587 github: https://handong1587.github.io/deep_learning/2015/10/09/object-detection.html

Minimal PyTorch implementation of YOLOv3

arXiv LaTeX Cleaner: Easily clean the LaTeX code of your paper to submit to arXiv

Code release for ConvNeXt model

[CVPR 2024] Real-Time Open-Vocabulary Object Detection