Stingray detection of aerial images using augmented training images generated by a conditional generative model
Journal
IEEE Computer Society Conference on Computer Vision and Pattern Recognition Workshops
Journal Volume
2018-June
Pages
1484-1490
Date Issued
2018
Author(s)
Abstract
In this paper, we present an object detection method that tackles the stingray detection problem based on aerial images. In this problem, the images are aerially captured on a sea-surface area by using an Unmanned Aerial Vehicle (UAV), and the stingrays swimming under (but close to) the sea surface are the target we want to detect and locate. To this end, we use a deep object detection method, faster RCNN, to train a stingray detector based on a limited training set of images. To boost the performance, we develop a new generative approach, conditional GLO, to increase the training samples of stingray, which is an extension of the Generative Latent Optimization (GLO) approach. Unlike traditional data augmentation methods that generate new data only for image classification, our proposed method that mixes foreground and background together can generate new data for an object detection task, and thus improve the training efficacy of a CNN detector. Experimental results show that satisfiable performance can be obtained by using our approach on stingray detection in aerial images. ? 2018 IEEE.
Subjects
Antennas; Computer vision; Object detection; Object recognition; Surface waters; Unmanned aerial vehicles (UAV); Data augmentation; Detection problems; Generative model; Object detection method; Sea surface area; Training image; Training sample; Training sets; Image enhancement
SDGs
Type
conference paper