Albumentations on GitHub in Drones and UAVs

#1

WangLibo1995/GeoSeg

UNetFormer: A UNet-like transformer for efficient semantic segmentation of remote sensing urban scene imagery, ISPRS. Also, including other vision transformers and CNNs for satellite, aerial image and UAV image segmentation.

#2

ZhaoJ9014/Anti-UAV

🔥🔥Official Repository for Anti-UAV🔥🔥

#3

iamaisim/ProjectAirSim

Project AirSim is Microsoft's evolution of AirSim, an advanced simulation platform for building, training, and testing autonomous systems in high-fidelity virtual environments

#4

kitoweeknd/RFUAV

This is official repository of our paper "RFUAV: A Benchmark Dataset for Unmanned Aerial Vehicle Detection and Identification". Codes include a two stage model to achieve drone detection and classification using some FFT/STFT analytical method. The Raw data will be free to use after our paper is accept. Star us!!!!, if you think this is useful♥

#6

wish44165/YOLOv12-BoT-SORT-ReID

🔥 CVPR 2025 (Nashville TN) - The 4th Anti-UAV Workshop & Challenge 🥉

#7

RoboticsKnowledgebase/roboticsknowledgebase.github.io

Robotics Knowledge Base. The Wiki for Robot Builders.

#8

Yux1angJi/GTA-UAV

[AAAI 2025 Oral🚁] Game4Loc: A UAV Geo-Localization Benchmark from Game Data

#9

IIM-TTIJ/MVA2023SmallObjectDetection4SpottingBirds

This is the repository with the baseline code for the "Small Object Detection Challenge for Spotting Birds" at MVA2023.

#10

UAV-AVL/Benchmark

Benchmark for UAV Visual Localization under Low-altitude Multi-view Observation Condition

#11

dtc111111/Multi-Modal-UAV

We win the 1st place in UG2+, a task in CVPR 2024 UAV Tracking and Pose-Estimation Challenge.

#12

ayushdabra/drone-images-semantic-segmentation

Multi-class semantic segmentation performed on "Semantic Drone Dataset."

#13

adityachintala/UAV-tracking-tank

MILITARY TANK OBJECT DETECTION AND TRACKING USING UAV

#14

forever208/yolov5_train_on_UAVDT

train the latest YOLOv5 with drone dataset UAVDT

#15

GXNU-ZhongLab/SGLATrack

Similarity-Guided Layer-Adaptive Vision Transformer for UAV Tracking (CVPR 2025)

#16

ISChenawei/DMNIL

[TNNLS'2026] Without Paired Labeled Data: End-to-End Self-Supervised Method for Drone-View Geo-Localization

#17

Irisky123/YOLOMG

Dataset and Codes for the submission paper "YOLOMG: Vision-based Drone-to-Drone Detection with Appearance and Pixel-Level Motion Fusion"

#18

MingboHong/SSPNet

SSPNet: Scale Selection Pyramid Network for Tiny Person Detection from UAV Images.

#19

chengshuxiao/YOLOv5-ODConvNeXt

YOLOv5-ODConvNeXt is an improved version of YOLOv5 for ship detection on drone-captured images.

#20

DanielRossi1/TakuNet

TakuNet: an Energy-Efficient CNN for Real-Time Inference on Embedded UAV systems in Emergency Response Scenarios

#21

we1pingyu/DSHNet

[WACV 2021] Towards Resolving the Challenge of Long-tail Distribution in UAV Images for Object Detection.

#22

NNNNerd/mmdet-rgbtdroneperson

Official code for "Drone-based RGBT Tiny Person Detection".

#23

nia194/Object-Detection

Object-Detection From Drone Surveillance (VisDrone Dataset)

#24

vt-le/HSTforU

This is an official implement for "HSTforU: Anomaly Detection in Aerial and Ground-based Videos with Hierarchical Spatio-Temporal Transformer for U-net"

#25

zsniko/EDNet

Implementation of EDNet: Edge-Optimized Small Target Detection in UAV Imagery (UIC'24)

#26

sunzc-sunny/refdrone

RefDrone: A Challenging Benchmark for Drone Scene Referring Expression Comprehension

#27

liukejia121/bearinguav

Bearing-UAV: Vision-Only Cross-View Pose Regression for UAV Navigation

#28

anhuipl2010/SOD-YOLO

SOD-YOLO (Small Object Detection YOLO) builds upon the foundational YOLOv8 model to address the unique challenges of detecting small objects in complex backgrounds typical of UAV imagery. This repository contains the code for the model, including the novel enhancements we have introduced.

#29

Annzstbl/MMOT

MMOT: The First Challenging Benchmark for Drone-based Multispectral Multi-Object Tracking[NIPS 2025]

#30

earth-insights/Dist-Tracker

The core technology of the international championship team in ​Track 3 (Multi-UAV Complex Scene Tracking)​​ of the ​4th Anti-UAV Challenge at CVPR

#31

2178568050/AntiUAV_Improved_Based_on-YOLOv11

复现南京理工大学一篇论文：A Lightweight Anti-Unmanned Aerial Vehicle Detection Method Based on Improved YOLOv11

#32

rs-dl/MOPAD

Dataset and codes of paper "Growing status observation for individual oil palm trees using Unmanned Aerial Vehicle (UAV) images"

#33

1e12Leon/UAV-LODBoost

A User-Friendly Toolkit for UAV Light-weighting Object Detection

#34

Bigtuo/Drone-YOLO

复现Drone-YOLOv8s，论文三明治结构中DW卷积核存在疑点，均改为3*3.

#35

grimmlab/UAVWeedSegmentation

Deep Learning-based Early Weed Segmentation using Motion Blurred UAV Images of Sorghum Fields

#36

StanfordMSL/SousVide

Cooking Drone Visuomotor Policies in a Gaussian Splat Vacuum

#37

rudrakshkapil/Integrated-RGB-Thermal-Orthomosaicing

Improved thermal orthomosaicking workflow using co-acquired RGB and thermal drone imagery. Published as Featured MDPI Journal Article.

#38

open-forest-observatory/geograypher

Enabling Geospatial Predictions from Individual Drone Images

#39

albertstarfield/project-zephyrine

Zephyrine: An augmented GNC framework system for experimental UAVs. Engineered for high-speed processing of navigation loops and sensor fusion and adaptation to bridge the gap between strategic virtual path and action planning and real-world kinematic control.

#41

ccccwb/Multimodal-Detection-and-Tracking-UAV

A Multimodal Detection and Tracking System based on DJI Payload SDK and Mobile SDK.

#43

connorlee77/uav-thermal-water-segmentation

Code for near-shore thermal water segmentation on UAVs. Datasets for network training included.

#44

microsoft/MegaDetector-Overhead

MegaDetector-Overhead — The Microsoft open-source AI for overhead wildlife detection. Point-based detection model for aerial and drone imagery, identifying wildlife from above. Maintained by Microsoft AI for Good Lab. Part of the Pytorch-Wildlife ecosystem.

#45

RoblabWh/argus

Analyse and view your UAV mapping flights in a structured webapp

#46

wxy16/SSFNet

SSFNet：Lightweight real-time network for semantic segmentation of UAV remote sensing images

#47

AmzadHossainrafis/DroneImage-segmentation

This is a PyTorch-based project for drone image segmentation. The goal of this project is to segment objects and regions of interest within aerial images captured by drones. Image segmentation is a crucial task in computer vision and has various applications, including agriculture, urban planning, and environmental monitoring.

#48

norlab-ulaval/droneSegmentation

Nasiri et al., Using Citizen Science Data as Pre-Training for Semantic Segmentation of High-Resolution UAV Images for Natural Forests Post-Disturbance Assessment, Forests, 2025

#49

ablanco1950/Drone-Detection_Yolov10

From dataset https://universe.roboflow.com/drone-detection-pexej/drone-detection-data-set-yolov7/dataset/1 a model is obtained, based on yolov10 to detect drones in images. Predictions from several models are used in cascade to obtain the optimal result.

#50

starkLJ/LGNet

Expert Systems With Applications：Shooting condition insensitive unmanned aerial vehicle object detection

#51

z00bean/HighwayTrafficAnalysis

Truck Rollover Crashes on Ramps. Funding agency- MassDOT

#52

up-count/uav-dot

The repository is a supplement to the paper "Enhancing people localisation in drone imagery for better crowd management by utilising every pixel in high-resolution images"

#54

Jinlab-AiPhenomics/OSNet

An oriented instance segmentation network of breeding plot extraction from UAV RGB imagery

#55

DVA-LAB/DVA_LAB_MVP

Drone Video Analysis(DVA) project for helping MARC supported by Kakao Impact, MODULABS

#56

moshenh01/Semantic-Template-Matching-for-UAV-Localization

Semantic Template Matching For Map-Based UAV Localization

#57

StyNW7/UPH_DJITello

Have fun with DJI Tello Drone by exploring every idea and possibility that can done with the drone!

#58

AlfonsoLRz/VineyardUAVClassification

Classification of hyperspectral UAV swaths using Convolutional Neural Networks.

#60

LittleFish-Coder/gen-ai-uav

AI CUP 2024 Spring - Rank13

#61

ausalsaka/UnityDroneSimulation

Unity project to simulate drone autonomous navigation via AI object recognition.

#62

BobErgot/Hazard-View-Bird

Efficient model for semantic segmentation on edge devices, specifically targeting the analysis of disaster scenes from images captured by unmanned aerial vehicles (UAVs).

#64

Charbel199/oil-spill-thickness-estimation

A proof of concept for using machine learning with radar data from a UAV over a body of water to estimate the thicness of an oil spill.

#66

naslab-projects/Aerial-Fluvial-Semantic-Segmentation

Semantic segmentation of fluvial images taken by drone

#67

nicolevgarduno/Waste-Detection-Drone

UAB's Cs 420/520 course

#68

yahavmh/My_VGG16

Using VGG16 CNN for weed detection on RGB images taken by UAV

#74

aj33b/aerial-satellite-imagery-segmentation-nepal

Aerial Satellite Imagery Segmentation Nepal is a project for my thesis of Master of Technology in Artificial Intelligence, Kathmandu University. The project contains implementation of several models to train on a custom drone image dataset of Nepal. The project tries to study the landcover segmentation with classes like buildings & woodlands.

#76

Amine-Fathallah0/LacunaAmine

Solar panel and boiler counting model for the Lacuna Solar Survey Challenge on Zindi. Built using InceptionV3, hybrid Albumentations + TorchVision augmentations, and trained with MAE loss. Final predictions made on satellite and drone imagery from Madagascar.

#80

burakkagann/drone-obj-detection-lightweight-ai

Robust Object Detection for Surveillance Drones in Low-Visibility Environments Using Lightweight AI Models

#82

deeprightai/TakuNet

TakuNet: an Energy-Efficient CNN for Real-Time Inference on Embedded UAV systems in Emergency Response Scenarios

#83

jamesahou/Drone-2-3-4-5

Drone adaptation of One-2-3-4-5

#86

melvinczyk/Waste-detection-drone

A Tello drone designed application powered by YOLO that allows users to detect and document waste.

#87

merryHunter/drone-tiles-mapping

Identify drone GPS by mapping aerial footage to Google map tiles

#88

OpenTrafficCam/tph-yolov5

Implementation of "TPH-YOLOv5: Improved YOLOv5 Based on Transformer Prediction Head for Object Detection on Drone-Captured Scenarios"

#89

pastoruri/utec-alberta-drone-models

Project Models

#91

phomarkon/mit-hackathon-2025-drone

MIT AI Global Hackathon Case 5

#93

rm2077/ADDS

Autonomous Drone Delivery System

#95

SANKETPise95/GreenScan-UAV

This repository provides a comprehensive approach to weed segmentation using UAV images, encompassing data preparation, model training, prediction, and evaluation.

#98

uci-uav-forge/uavf_2023

UAV Forge's imaging and navigation repository for the 2022-23 school year