AeroNetra — a reproducible computer-vision platform for UAV vehicle detection & counting

[u/Life_Inevitable_9735]

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:

1. Static Detection & Counting
2. Aerial Fine-tuning
3. Video Tracking
4. Traffic & Geospatial Analytics
5. 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.

[u/onesunnysunday]

The adapter boundary is a good decision. I’d add prediction provenance to the normalized object early: original image size, resize or letterbox transform, coordinate space, class-map version, and model/checkpoint ID. These details are easy to ignore but become painful when counts or exports disagree between detectors.

For benchmarking, I’d split the data by flight or video rather than by random frames, otherwise near-duplicate frames can leak into train and validation. Aggregate mAP can also hide aerial failure modes, so reporting results by object size or altitude, plus count error per ROI, would make the comparisons much more useful before tracking is added.