STMG-PCNet

STMG-PCNet: Explainable Lightweight Shape-Texture Mutual-Guided Predictor-Corrector Network for Infrared Small Target Detection

Zichen Zhao, Member, IEEE, Huanzhang Ling, Member, IEEE, Xiaxia Qin, Haodong Xiao, Jihao Wu, Chengwei Liu,

This is the official implementation of the 'STMG-PCNet: Explainable Lightweight Shape-Texture Mutual-Guided Predictor-Corrector Network for Infrared Small Target Detection' .

🔥 use STMG-PCNet with only one file !

python train2.py
python test2.py

Remember to change the path of dataset into your own!

Import the right path of model.

And configure the correct Python library.

Overview