Skip to content
No description, website, or topics provided.
Shell
Branch: master
Clone or download
Latest commit 8d30fe3 Jul 5, 2019
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
center_loss rm2 Jul 5, 2019
siamese AIC init Jul 3, 2019
README.md rm2 Jul 5, 2019

README.md

Successive Embedding and Classification Loss for Aerial Image Classification

[Paper]

Overview

This is the CAFFE implementation of "Successive Embedding and Classification Loss for Aerial Image Classification".

Requirements

[CAFFE with Center Loss]

Data Preparation

  1. Download [MSTAR dataset]. Organize the subset of the dataset following Table I of the paper;

  2. Randomly partition the training set into real training set (with name "train90_sqrt_mag_aug.h5") and validation set (with name "train10_sqrt_mag_aug.h5"). Name the test set "test_sqrt_mag_aug.h5".

Run the Siamese Network

  1. Run grid_siamese.sh (note that this will automatically run batches of experiments. Change hyperparameter if you only want to run one or some experiments with specific hyperparameters.)

  2. Note that the prototxt of contrastive loss on feature layer can be found at "siamese_feature.prototxt".

Run the Network with Center Loss

  1. Run grid.sh and will grid learning rate for network trained with center loss on classifier layer.

License and Citation

The use of this software is RESTRICTED to non-commercial research and educational purposes.

You can’t perform that action at this time.