Skip to content

High quality sample generation with self-attention adaptive GAN

Notifications You must be signed in to change notification settings

kdxiaozhi/SaGAAN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

SaGAAN aims to generate realistic hyperspectral profiles by feeding a small amount of labeled samples. It is developed to alleviate the problem of sample shortage when performing hyperspectral image classification. There are two contributions in this work, being 1) self-attention is introduced to waive unwanted noises, and 2) domain adaptation reinforced the similarity of generated samples. The SaGAAN not only helpful in hyperspectral data analysis domain, it is possible to be useful in time-series data processing. Please cite this work if you find it interesting.

"Sample Generation with Self-Attention Generative Adversarial Adaptation Network (SaGAAN) for Hyperspectral Image Classification"

Note: To use the SaGAAN code, you may want to make sure the necessary environment already satisfied. The related modules or packages are:

  • tensorflow
  • matplotlib
  • scipy
  • etc...

How to use: You should aware this code is in its prototype format, you may need to revise parameters according to your datasets or tasks. Here is the general overview of this project.

Run "SaGAAN.py" to generate samples

You should make sure all files are in right positions.

  1. check lines 168, 227-228, make sure all training files are ready;
  2. check generator and discriminator have the correct setting (here is Pavia dataset config)
  3. check line 384, make sure the output samples is in right file

At last you should plot the generated samples and compare them with the original ones, like this SaGAAN

About

High quality sample generation with self-attention adaptive GAN

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages