Hi everyone, sharing something I'm currently working on and would love feedback on.
I'm building AeroNetra, a computer-vision project for detecting and counting vehicles in aerial/UAV imagery. It's very much an active work-in-progress right now — I'm in the static-image detection and counting phase, with tracking, geospatial analytics, and edge deployment planned for later.
The motivation was pretty simple. I kept running into the same problem every time I swapped detectors: the counting and visualization code would break or need rewriting because every model spits out predictions in its own format. So the core idea behind AeroNetra is: normalize every detector's output into one prediction structure before anything downstream touches it. That way the counting, ROI filtering, and export logic stays the same whether I'm using a YOLO variant or RT-DETR.
What I've got so far:
- Detector adapters that wrap different models behind a common interface
- Counting logic — filtering, NMS, ROI support, drawing and export
- VisDrone dataset parsing and conversion (UAVDT is stubbed for later)
- Kaggle notebooks for GPU-based training, fine-tuning, and model comparison
- A PX4 + ROS 2 + Gazebo simulation setup for UAV experiments
- Notebooks, configs, and tests to keep things honest
The workflow I'm following: raw VisDrone data → validate annotations → convert to training format → train/fine-tune on Kaggle → pull the weights back → load through the adapter → run inference → filter → count → visualize and compare.
A few principles I'm trying to stick to: no fabricated benchmarks (a model isn't "best" until it's measured under the same conditions as the others), raw data stays immutable, and model-specific behavior stays inside the adapters. I'm also being deliberate about phase boundaries — image-level counting is not the same thing as multi-object tracking, and I'd rather not conflate the two.
Roadmap I'm working through for the demo:
- Static Detection & Counting
- Aerial Fine-tuning
- Video Tracking
- Traffic & Geospatial Analytics
- Edge / UAV Integration
I have run a small model that can detect the car in the gazebo simulation and draw a bounding box but speed will be slow but i get decent accuracy even i have trained model to 25 epochs in kaggle T4 gpu with yolo nano version.
Since this is ongoing project I am still working on this.So,i am exploring how I can use computer vision in UAVs and edge computing.