Stars
D2Go is a toolkit for efficient deep learning
Code used to generate synthetic scenes and bounding box annotations for object detection. This was used to generate data used in the Cut, Paste and Learn paper
Refine high-quality datasets and visual AI models
Torchreid: Deep learning person re-identification in PyTorch.
BoxMOT: pluggable SOTA tracking modules for segmentation, object detection and pose estimation models
NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source components of TensorRT.
This repository contains code for object tracking in videos using the YOLO-NAS object detection model and the DeepSORT algorithm.
BoT-SORT: Robust Associations Multi-Pedestrian Tracking
Easily train or fine-tune SOTA computer vision models with one open source training library. The home of Yolo-NAS.
"Retinexformer: One-stage Retinex-based Transformer for Low-light Image Enhancement" (ICCV 2023) & (NTIRE 2024 Challenge)
Image-to-Image Translation in PyTorch
Software that can perform photorealistic style transfer without the need of any post-processing steps.
Cut and paste augmentation for object detection and instance segmentation
Object tracking implemented with the Roboflow Inference API, DeepSort, and OpenAI CLIP.
[ECCV 2024] Official implementation of the paper "Grounding DINO: Marrying DINO with Grounded Pre-Training for Open-Set Object Detection"
Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125
Annotate better with CVAT, the industry-leading data engine for machine learning. Used and trusted by teams at any scale, for data of any scale.
OpenMMLab Detection Toolbox and Benchmark
Simple tool to split COCO annotations into train/test datasets.
Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.
Publication-ready NN-architecture schematics.
A Python wrapper for the Chemistry Development Kit (CDK)
Fit interpretable models. Explain blackbox machine learning.
Quantus is an eXplainable AI toolkit for responsible evaluation of neural network explanations