Awesome

SAFINet

Network Architecture

image text

Requirements

python 3.7 + pytorch 1.9.0

Saliency maps

We provide saliency maps of our SAFINet on ORSSD (code:SAFI) and EORSSD (code:SAFI) datasets.

Training

Run train_SAFINet.py.

Pre-trained model and testing

Download the following pre-trained model and put them in ./models/SAFINet/, then run test_SAFINet.py.
SAFINet_ORSSD (code:SAFI)
SAFINet_EORSSD (code:SAFI)

Evaluation Tool

You can use the evaluation tool (MATLAB version) to evaluate the above saliency maps.