Awesome
CorrNet
This project provides the code and results for 'Lightweight Salient Object Detection in Optical Remote Sensing Images via Feature Correlation', IEEE TGRS, vol. 60, pp. 1-12, 2022. IEEE link and arxiv link Homepage
Network Architecture
<div align=center> <img src="https://github.com/MathLee/CorrNet/blob/main/image/CorrNet.png"> </div>Accuracy v.s. Parameters
<div align=center> <img src=https://github.com/MathLee/CorrNet/blob/main/image/accuracyVSparams.png width=52% /> </div>Requirements
python 2.7 + pytorch 0.4.0 or
python 3.7 + pytorch 1.9.0
Saliency maps
We provide saliency maps and measure results (.mat) (code: m1dm) of all compared methods (code: kftm) and our CorrNet (code: fbee) (or under './saliencymap/') on ORSSD and EORSSD datasets.
In addition, we also provide saliency maps of our CorrNet on the recently published ORSI-4199 dataset under './saliencymap/'.
Training
Modify paths of VGG backbone (code: ego5) in /model/vgg.py and datasets, then run train_CorrNet.py.
Pre-trained model and testing
Download the following pre-trained model, and modify paths of pre-trained model and datasets, then run test_CorrNet.py.
We also uploaded these pre-trained models in /models.
ORSSD (code: vqi7)
EORSSD (code: q5mr)
ORSI-4199 (code: va3b)
Evaluation Tool
You can use the evaluation tool (MATLAB version) to evaluate the above saliency maps.
ORSI-SOD_Summary
Citation
@ARTICLE{Li_2022_CorrNet,
author = {Gongyang Li and Zhi Liu and Zhen Bai and Weisi Lin and Haibin Ling},
title = {Lightweight Salient Object Detection in Optical Remote Sensing Images via Feature Correlation},
journal = {IEEE Transactions on Geoscience and Remote Sensing},
volume = {60},
pages = {1-12},
year = {2022},
}
If you encounter any problems with the code, want to report bugs, etc.
Please contact me at lllmiemie@163.com or ligongyang@shu.edu.cn.