CoolFace
Datasetpublic

rollingcoconut/drone-detection-assignment3

UAV Drone Detection and Tracking Detect and track drones in video using YOLO fine-tuned detector, Kalman Filter w/ tuned noise parameters. Source Videos Video URL drone_video_1 YouTube - DhmZ6W1UAv4 drone_video_2 YouTube - YrydHPwRelI Training Dataset Dataset: lgrzybowski/seraphim-drone-detection-dataset Training images used: 33296 (from ~77K ) Test images: 8349 Detector Model: YOLOv8n fine-tuned on Seraphim… See the full description on the dataset page: https://huggingface.co/datasets/rollingcoconut/drone-detection-assignment3.

sourceHugging Faceupdated 6mo agoView on Hugging Face
0likes10downloads
Dataset Card

UAV Drone Detection and Tracking

Detect and track drones in video using YOLO fine-tuned detector, Kalman Filter w/ tuned noise parameters.

Source Videos

VideoURL
dronevideo1YouTube - DhmZ6W1UAv4
dronevideo2YouTube - YrydHPwRelI

Training Dataset

Detector

  • —Model: YOLOv8n fine-tuned on Seraphim Drone Detection Dataset
  • —Training: 15 epochs, 640x640, batch 16
  • —Classes: 1 (drone)
  • —Performance: Precision 0.93, Recall 0.863
  • —Frame extraction: 5 fps via ffmpeg

Schema

detections — Frame-Level Drone Detections

One row per video frame that contains at least one drone detection. Confidence threshold >= 0.5.

ColumnTypeDescription
videostringVideo name (dronevideo1, dronevideo2)
frame_indexintFrame number within the video (0-indexed at 5 fps)
imagebytesThe video frame image (JPEG)
num_detectionsintNumber of drones detected in this frame
detectionsstring (JSON list)List of detections with bbox and confidence

Example detections value:

json
[{"bbox": [839.67, 173.83, 919.57, 229.07], "confidence": 0.8177}]

NOTE: [x_min, y_min, x_max, y_max] pixel coordinates.

Detection Summary

VideoTotal FramesFrames with DetectionsDetection Rate
dronevideo182855166.5%
dronevideo22,5801786.9%