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
pip install --upgrade mxnet-cu100 gluoncv
The trained model params and training logs are in ./params
The training commands / shell scripts are in cmd_scripts.txt
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}
}