Home

Awesome

ACM-pytorch

The unofficial implement of paper "Asymmetric Contextual Modulation for Infrared Small Target Detection" in Pytorch. And this is my Homepage.

More Official Information:

Official code: open-acm and Homepage of Yimian Dai

Dependencies

Python = 3.6
Pytorch = 1.2.0

Train Command

python train.py --backbone-mode FPN --fuse-mode BiLocal
python train.py --backbone-mode FPN --fuse-mode AsymBi
python train.py --backbone-mode FPN --fuse-mode BiGlobal
python train.py --backbone-mode UNet --fuse-mode BiLocal
python train.py --backbone-mode UNet --fuse-mode AsymBi
python train.py --backbone-mode UNet --fuse-mode BiGlobal

Experiment Results

Net ModeBest IoUBest nIoU
FPN+BiLocal0.66750.6781
FPN+AsymBi0.69540.7121
FPN+BiGlobal0.69180.7010
UNet+BiLocal0.64020.6800
UNet+AsymBi0.71520.7281
UNet+BiGlobal0.69750.7227