OCOR

This is a reproduction of the CVPR'22 paper, Bi-directional Object-context Prioritization Learning for Saliency Ranking.

Installation

step 1: install pytorch and mmcv==1.3.9 referring to this.

step 2: clone this repository and execute:

python setup.py develop

step 3: install apex following this. (optional)

Dataset

Download COCO-style JSON files for the ASSR dataset from: https://pan.baidu.com/s/1XvYwBCn3sc6lAlJbJ94gUQ (pwd: ocor)

Training

bash tools/dist_train.sh configs/ocor/ocor_swin...py num_gpus

Inference & Evaluation

python inference.py

python evaluate_SOR.py

Pre-trained Model

Download it from our Baidu cloud: https://pan.baidu.com/s/15tINLiVC8kPQm6xqxyaJlA (pwd: ocor), then use it for fine-tuning, and inference.

Results

Visual results for ASSR dataset: https://pan.baidu.com/s/1V3MzSBWI_5UNSfC1YCVf3g (pwd: ocor)

Citation

@inproceedings{tian2022bi,
  title={Bi-directional object-context prioritization learning for saliency ranking},
  author={Tian, Xin and Xu, Ke and Yang, Xin and Du, Lin and Yin, Baocai and Lau, Rynson WH},
  booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
  pages={5882--5891},
  year={2022}
}

GrassBro/OCOR