A Generative Approach Towards Improved Robotic Detection of Marine Litter
arXiv:1910.04754
Abstract
This paper presents an approach to address data scarcity problems in underwater image datasets for visual detection of marine debris. The proposed approach relies on a two-stage variational autoencoder (VAE) and a binary classifier to evaluate the generated imagery for quality and realism. From the images generated by the two-stage VAE, the binary classifier selects "good quality" images and augments the given dataset with them. Lastly, a multi-class classifier is used to evaluate the impact of the augmentation process by measuring the accuracy of an object detector trained on combinations of real and generated trash images. Our results show that the classifier trained with the augmented data outperforms the one trained only with the real data. This approach will not only be valid for the underwater trash classification problem presented in this paper, but it will also be useful for any data-dependent task for which collecting more images is challenging or infeasible.
References in corpus (10)
- Conditional Generative Adversarial Nets
- GANs Trained by a Two Time-Scale Update Rule Converge to a Local Nash Equilibrium
- Large Scale GAN Training for High Fidelity Natural Image Synthesis
- The Effectiveness of Data Augmentation in Image Classification using Deep Learning
- Data Augmentation Generative Adversarial Networks
- Few-Shot Learning with Graph Neural Networks
- Generative Adversarial Networks in Computer Vision: A Survey and Taxonomy
- Neural Photo Editing with Introspective Adversarial Networks
- Wasserstein Auto-Encoders
- Diagnosing and Enhancing VAE Models