← all repositories

Media-Smart/vedadet

A PyTorch object detection toolbox for single-stage detectors like RetinaNet and FCOS with TensorRT model conversion.

503 stars Python Computer VisionML Frameworks
vedadet
Velocity · 7d
+0.2
★ / day
Trend
steady
star history

Vedadet is a modular single-stage object detection framework built on PyTorch. It decomposes detectors into four parts: data pipeline, model, postprocessing and criterion for easier deployment. The toolbox supports popular single-stage architectures including RetinaNet and FCOS, with built-in support for converting models to TensorRT engines for efficient inference on NVIDIA hardware.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.