Master Thesis:
Generative Adversarial Networks for Real-robot Missions

Informations

Author: Pin-Wei "David" Chen
E-mail: ccpwearth@gmail.com
Code: https://github.com/championway/gan_rv_thesis
Thesis: PDF

Abstract

Leveraging highly developed deep learning and artificial intelligence, computer vision technology and applications reached new levels. Computers can now not only perform image processing, classification, and object detection, but also can ”create” images similarly to humans, due to generative model developments. In particular, the generative adversarial network (GAN) provides many architectures and applications, such as image style transfer, human face generation, image generation from text, etc. However, there has been little study regarding applying GAN to real-robot missions to replace and improve other approaches. Therefore, this work proposed two GANs: FCN-Pix2Pix and SSIM-CycleGAN, based on Pix2Pix and CycleGAN respectively, and implemented them for two real-robot missions which still face some challenges with modern solutions: semantic segmentation and virtual dataset from sim to real. The proposed approaches were also compared with current state-of-the-art approaches, verifying significant advantages for the proposed methods.

Image

Network Architectures

fcn_gan SSIM-CycleGAN

Experiments

ex1 ex2

Tote Sim2Real Dataset

  • Class : 2
  • Class list : Real totes, Unity totes
  • Image size : 640*480
  • Image Description :
    • Real tote : A group of tote images photoed by SR300 from real world with different view angles and distances
    • Unity tote : A group of tote images generated from virtual environment (Unity) with different view angles and distances
    • Image : Unity tote images with objects inside. Every image has a corresponding labeled mask in mask folder
    • Mask : Labeled masks from Unity RGB images
  • CSV Description :
    • menu.csv : It shows the mapping between Unity image (column 1), mask (column 2) and Unity tote image (column 3).
    • real_totes_train.csv : A list of real totes image for training
    • real_totes_test.csv : A list of real totes image for testing
    • unity_totes_train.csv : A list of Unity totes image for training
    • unity_totes_test.csv : A list of Unity totes image for testing
Dataset Size Image Real totes Unity totes Dataset(.tar)
Tote Sim2Real Sample 25.9MB 100 images 100 images 99 images sim2real_sample.zip
Tote Sim2Real 19.8GB 68608 images 5690 images 9472 images sim2real.zip