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

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

#3

ZhaoJ9014/Anti-UAV

🔥🔥Official Repository for Anti-UAV🔥🔥

#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, USA) - The 4th Anti-UAV Workshop & Challenge 🥉

#7

UAV-AVL/Benchmark

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

#8

RoboticsKnowledgebase/roboticsknowledgebase.github.io

Robotics Knowledge Base. The Wiki for Robot Builders.

#9

Yux1angJi/GTA-UAV

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

#10

IIM-TTIJ/MVA2023SmallObjectDetection4SpottingBirds

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

#11

zsniko/EDNet

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

#12

GXNU-ZhongLab/SGLATrack

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

#13

dtc111111/Multi-Modal-UAV

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

#14

z637826/yolo-omni

YOLO-OMNI: A cross-domain real-time object detection framework for fisheye, drone, panorama, game-to-real, and mixed-camera scenarios.

#15

ayushdabra/drone-images-semantic-segmentation

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

#16

adityachintala/UAV-tracking-tank

MILITARY TANK OBJECT DETECTION AND TRACKING USING UAV

#17

wd-sir/UAVDETR

This is the official code for UAV-DETR: DETR for Anti-Drone Target Detection.

#18

forever208/yolov5_train_on_UAVDT

train the latest YOLOv5 with drone dataset UAVDT

#19

ISChenawei/DMNIL

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

#20

Irisky123/YOLOMG

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

#21

liukejia121/bearinguav

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

#22

MingboHong/SSPNet

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

#23

NNNNerd/mmdet-rgbtdroneperson

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

#24

chengshuxiao/YOLOv5-ODConvNeXt

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

#25

we1pingyu/DSHNet

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

#26

DanielRossi1/TakuNet

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

#27

YaxuanLi-cn/PairUAV

📌 Official baseline implementation of 'Last-Meter Precision Navigation for UAVs: A Diffusion-Refined Aerial Visual Servoing Approach‘, serving as the baseline for the UAVM @ ACM MM 2026 Workshop Challenge.

#28

nia194/Object-Detection

Object-Detection From Drone Surveillance (VisDrone Dataset)

#29

2178568050/AntiUAV_Improved_Based_on-YOLOv11

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

#30

1e12Leon/UAV-LODBoost

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

#31

grimmlab/UAVWeedSegmentation

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

#32

StanfordMSL/SousVide

Cooking Drone Visuomotor Policies in a Gaussian Splat Vacuum

#33

rudrakshkapil/Integrated-RGB-Thermal-Orthomosaicing

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

#34

open-forest-observatory/geograypher

Enabling Geospatial Predictions from Individual Drone Images

#36

ccccwb/Multimodal-Detection-and-Tracking-UAV

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

#38

connorlee77/uav-thermal-water-segmentation

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

#39

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.

#40

RoblabWh/argus

Analyse and view your UAV mapping flights in a structured webapp

#41

wxy16/SSFNet

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

#42

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.

#43

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

#44

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.

#45

starkLJ/LGNet

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

#46

z00bean/HighwayTrafficAnalysis

Truck Rollover Crashes on Ramps. Funding agency- MassDOT

#47

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"

#49

Jinlab-AiPhenomics/OSNet

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

#50

DVA-LAB/DVA_LAB_MVP

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

#51

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

Semantic Template Matching For Map-Based UAV Localization

#52

StyNW7/UPH_DJITello

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

#53

AlfonsoLRz/VineyardUAVClassification

Classification of hyperspectral UAV swaths using Convolutional Neural Networks.

#55

LittleFish-Coder/gen-ai-uav

AI CUP 2024 Spring - Rank13

#56

ausalsaka/UnityDroneSimulation

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

#57

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).

#59

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.

#61

naslab-projects/Aerial-Fluvial-Semantic-Segmentation

Semantic segmentation of fluvial images taken by drone

#62

nicolevgarduno/Waste-Detection-Drone

UAB's Cs 420/520 course

#63

yahavmh/My_VGG16

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

#68

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.

#70

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.

#74

burakkagann/drone-obj-detection-lightweight-ai

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

#76

deeprightai/TakuNet

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

#77

jamesahou/Drone-2-3-4-5

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

#80

melvinczyk/Waste-detection-drone

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

#81

merryHunter/drone-tiles-mapping

Identify drone GPS by mapping aerial footage to Google map tiles

#82

OpenTrafficCam/tph-yolov5

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

#83

pastoruri/utec-alberta-drone-models

Project Models

#85

phomarkon/mit-hackathon-2025-drone

MIT AI Global Hackathon Case 5

#87

rm2077/ADDS

Autonomous Drone Delivery System

#89

SANKETPise95/GreenScan-UAV

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

#92

uci-uav-forge/uavf_2023

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