Date of Award

3-2022

Document Type

Thesis

Degree Name

Master of Science

Department

Department of Operational Sciences

First Advisor

Bruce A. Cox, PhD

Abstract

A challenging task in computer vision is finding techniques to improve the object detection and classification capabilities of ML models used for processing images acquired by moving aerial platforms. This research explores if GAN augmented UAV training sets can increase the generalizability of a detection model trained on said data. To answer this question, the YOLOv4-Tiny Object Detection Model was trained with aerial image training sets depicting rural environments. The salient objects within the frames were recreated using various GAN architectures, placed back into the original frames, and the augmented frames appended to the original training sets. GAN augmentation on aerial image training sets led to a 6.75% increase on average in the mAP of the YOLOv4-Tiny Object Detection model with a best-case increase of 15.76%. Similarly, a 4.13% increase on average and a best-case increase of 9.60% was observed for the IoU rate. Finally, 100.00% TP, 4.70% FP, and zero FN detection rates were yielded, providing further evidence supporting GAN augmentation for object detection model training sets.

AFIT Designator

AFIT-ENS-MS-22-M-151

DTIC Accession Number

AD1172373

Share

COinS