Code Monkey home page Code Monkey logo

featurehallucination-cgan's Introduction

Feature Hallucination Using Conditional GAN

Implementation of the paper - An Adversarial Approach to Discriminative Modality Hallucination for Remote Sensing Data - IEEE ICCV - CROMOL 2019 in TensorFlow

Uses a conditional GAN with novel discriminator of 2C classes (corresponsing to C real and fake classes each) to hallucinate missing features for hyperspectral or multimodal remote sensing data using available sensor derived features.

Generator takes latent dimensions as well as conditional input from available modality to generate missing features. GAN was trained using tips from Soumith Chintala's GANHacks.

Deep CNNs have been utilized as feature extractors for application of GAN, and further classification. Hallucinated features are concatenated with original features and sent to DNN classifier.

Ablation studies done with Indian Pines dataset.

Poster and Architecture

Poster Presented

t-SNE Plots

Without sensor abnormality With missing bands due to failures

Datasets

Results

Train Test Split Experiments

(Temperature 2, 50-50 Band Split)

Split 75-25 50-50 25-75 10-90
Teacher Network 97.54194303550526% 90.71219512195122% 70.28749837697956% 69.51761517809012%
Student Network 93.79633242294186% 92.13658536701668% 80.57759854532085% 71.92411924248465%

Temperature Experiments

(25-75 Train Test Split, 50-50 Band Split)

Temperature 1 2 3 5 10
Teacher Network 69.4289059482037% 85.625081307652% 72.19981788053645% 81.63132561622491% 57.47365682630959%
Student Network 77.31234552461086% 95.4728762846364% 79.84909587848074% 84.75347990733494% 64.39443216129073%

Band Split Experiments (Hallucinated-Real)

(Temperature 2, 25-75 Train Test Split)

Split 25-75 50-50 75-25 90-10
Teacher Network 55.60036425191497% 85.625081307652% 62.29998699334999% 64.70664758799801%
Student Network 56.70612723088876% 95.4728762846364% 86.09340444906985% 81.43619097332131%

Usage

Run the files in following order, modifying the directory of dataset -

python twostreamdistillation.py
python train.py

The file testmodels.py is used to evaluate the optimal model for generator.

python testmodels.py
python test.py

Citation

If this paper and/or this code is useful for your research, please consider citing us -

@inproceedings{pande2019adversarial,
  title={An Adversarial Approach to Discriminative Modality Distillation for Remote Sensing Image Classification},
  author={Pande, Shivam and Banerjee, Avinandan and Kumar, Saurabh and Banerjee, Biplab and Chaudhuri, Subhasis},
  booktitle={Proceedings of the IEEE International Conference on Computer Vision Workshops},
  pages={0--0},
  year={2019}
}

featurehallucination-cgan's People

Contributors

theavicaster avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Forkers

guoshuaixu

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.