SCWSSOD
This is the implementation of Structure-Consistent Weakly Supervised Salient Object Detection with Local Saliency Coherence (AAAI2021)
.
Training
Requirements
- Clone this project and install required pytorch first.
- pip install -r requirements.txt
Training data
The training data can be downloaed from Scribble_Saliency.
Pretrained weights for backbone
The pretrianed weight for backbone can be downloaded from Res50.
Traing procedure
- Download training data and put them into 'data' folder.
- run train.py
Testing
Test Model
The test model can be downloaded from model.
Testing procedure
- Modify test path
- run test.py
Predicted Maps
The predicted Saliency Maps can be downloaded from prediction.
Evaluation
The evaluation code is from GCPANet.
Others
The code is based on GCPANet and GatedCRFLoss.