NEW - YOLOv8 🚀 in PyTorch > ONNX > OpenVINO > CoreML > TFLite

Machine Learning Engineering Open Book

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

SpikingJelly is an open-source deep learning framework for Spiking Neural Network (SNN) based on PyTorch.

Deep and online learning with spiking neural networks in Python

Depth Pro: Sharp Monocular Metric Depth in Less Than a Second.

[CVPR 2024] Depth Anything: Unleashing the Power of Large-Scale Unlabeled Data. Foundation Model for Monocular Depth Estimation

Depth Anything V2. A More Capable Foundation Model for Monocular Depth Estimation

The open-source tool for building high-quality datasets and computer vision models

Effortless data labeling with AI support from Segment Anything and other awesome models.

An open source implementation of CLIP.

The repository provides code for running inference with the Meta Segment Anything Model 2 (SAM 2), links for downloading the trained model checkpoints, and example notebooks that show how to use th…

Lane Detection for CCTV-cameras using CLRNet: Cross Layer Refinement Network

Developing a state-of-the-art traffic surveillance system on edge devices for real-time information extraction under various weather conditions

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

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.

Implementation of paper - YOLOv9: Learning What You Want to Learn Using Programmable Gradient Information

Traffic Surveillance Data Generation capable of producing various environment record on road.

A Region-and-Trajectory Traffic Flow Estimation

eTram - Event-based Traffic Monitoring Dataset

PyTorch code and models for the DINOv2 self-supervised learning method.

A fast MoE impl for PyTorch

[ECCV 2022] ByteTrack: Multi-Object Tracking by Associating Every Detection Box

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

A playbook for systematically maximizing the performance of deep learning models.