/open-acm

code and trained models for "Asymmetric Contextual Modulation for Infrared Small Target Detection"

Primary LanguagePython

Asymmetric Contextual Modulation

MXNet/Gluon code for "Asymmetric Contextual Modulation for Infrared Small Target Detection" https://arxiv.org/abs/2009.14530

What's in this repo so far:

  • Code and trained models for the SIRST (Single-frame InfraRed Small Target detection) dataset

The SIRST dataset is available at https://github.com/YimianDai/sirst

Requirements

Install MXNet and Gluon-CV:

pip install --upgrade mxnet-cu100 gluoncv

Experiments

The trained model params and training logs are in ./params

The training commands / shell scripts are in cmd_scripts.txt

Citation

Please cite our paper in your publications if our work helps your research. BibTeX reference is as follows.

@inproceedings{dai21acm,
  title   =  {Asymmetric Contextual Modulation for Infrared Small Target Detection},
  author  =  {Yimian Dai and Yiquan Wu and Fei Zhou and Kobus Barnard},
  booktitle =  {{IEEE} Winter Conference on Applications of Computer Vision, {WACV} 2021}
  year    =  {2021}
}