dronefreak/VisDrone-dataset-python-toolkit
Modern PyTorch toolkit for the VisDrone aerial object detection dataset with production-ready training pipelines, real-time inference, and format converters. Features state-of-the-art models (Faster R-CNN, FCOS, RetinaNet), mixed precision training, rich progress tracking, and optimizations for small object detection in drone imagery.
This toolkit helps you analyze drone footage by automatically finding and labeling objects like cars, people, and bicycles in aerial images. You provide drone imagery (photos or video) and it outputs analyzed images with detected objects highlighted. This is ideal for anyone working with drone-collected data for surveillance, traffic monitoring, or environmental surveys who needs to identify specific items within that footage.
Use this if you need to accurately detect small objects like vehicles and pedestrians in drone-captured images and want a ready-to-use system to train and deploy detection models.
Not ideal if your primary use case involves detecting objects in standard ground-level photography or if you need to detect highly occluded objects or objects smaller than 15 pixels in aerial views.
Stars
61
Forks
22
Language
Python
License
Apache-2.0
Category
Last pushed
Mar 19, 2026
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/computer-vision/dronefreak/VisDrone-dataset-python-toolkit"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Related tools
lucidrains/vit-pytorch
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with...
roflcoopter/viseron
Self-hosted, local only NVR and AI Computer Vision software. With features such as object...
blakeblackshear/frigate
NVR with realtime local object detection for IP cameras
levan92/deep_sort_realtime
A really more real-time adaptation of deep sort
notAI-tech/NudeNet
Lightweight nudity detection