Open Source

Building in the open

We open-source our tools, datasets, and benchmarks to accelerate the computer vision ecosystem. Every CV team needs these — we're giving them away.

github.com/drik-vision

DrikLabel (Chitrā)

FEATURED
tools

AI-assisted video annotation tool with active learning propagation.

Auto-split video into segments, pre-annotate with YOLO/SAM, and propagate corrections across frames with active learning. Corrections auto-apply to past and future frames. Supports re-tracking of objects however small, in all directions, with classification updates.

annotation video active-learning +2
TypeScript

DrikSynth (Māyā)

FEATURED
tools

Unreal Engine based synthetic video data generator for training and validation.

Procedurally generate millions of life-like traffic videos with perfect ground truth labels. Configurable environments, weather, lighting, vehicle density, and camera angles. Built on Unreal Engine for photorealistic rendering.

synthetic-data unreal-engine procedural-generation +1
C++ / Python

DrikNetra

FEATURED
models

Multi-frame license plate super-resolution from video clips.

Fuses multiple video frames of a license plate bounding box to produce a single crisp, readable plate image. Based on BasicVSR++ / Real-ESRGAN / RVRT architectures. Solves the universal ANPR problem of blurry plates from surveillance cameras.

super-resolution ANPR license-plate +1
Python

drik-bench

benchmarks

Benchmark suite for evaluating AI models on Indian traffic scenarios.

Standardized evaluation framework for detection, tracking, ANPR, and scene understanding in Indian traffic conditions. Covers edge cases like three-wheelers, overloaded trucks, no-lane driving, and mixed traffic.

benchmark evaluation Indian-traffic +1
Python

Indian Plate Dataset

datasets

Diverse Indian license plate dataset with varied formats, fonts, and languages.

Curated dataset of Indian license plates covering all state codes, bilingual text, fancy fonts, dealer plates, government plates, and degraded conditions (blur, night, rain). Annotations include plate text, bounding box, and metadata.

license-plate India OCR +2
Dataset

Traffic Chaos 100

datasets

100 challenging Indian traffic scenes for model evaluation.

100 hand-picked, extremely challenging traffic scenes from Indian roads. Dense mixed traffic, unusual vehicles, poor visibility, complex intersections. Designed to stress-test detection and tracking models.

traffic evaluation edge-cases +1
Dataset

drik-track

models

Vehicle tracking toolkit optimized for Indian traffic conditions.

Multi-object tracking toolkit with re-identification, occlusion handling, and camera handoff. Tuned for dense, chaotic traffic environments where standard trackers fail.

tracking MOT vehicle +1
Python

drik-detect

models

Detection model configs and weights for Indian traffic objects.

Pre-trained detection models and configuration files covering 50+ Indian vehicle types including auto-rickshaws, tempos, tractors, bullock carts, and more. Ready-to-deploy configs for TensorRT and ONNX Runtime.

detection YOLO Indian-vehicles +1
Python
8 Repositories
2 Tools
2 Datasets
3 Models