A Multi-Task Collaborative Network for Light Field Salient Object Detection

April 6, 2022 ยท View on GitHub

Training stage

  1. Prepare the training data and change the data path
  2. Download the initial model - hqpw to the weights folder
  3. Train the model: python3 run_train.py --mode train

Testing stage

  1. Download the pretrained model- rrg7
  2. Download a example of testing data -9249
  3. Test the model: python3 test.py --mode test --sal_mode h --model ./models/mtcnet.pth

Our code is rewritten on the basis of this paper. Thanks a lot for the excellent code provided by the authors.

@inproceedings{zhao2019EGNet, title={EGNet:Edge Guidance Network for Salient Object Detection}, author={Zhao, Jia-Xing and Liu, Jiang-Jiang and Fan, Deng-Ping and Cao, Yang and Yang, Jufeng and Cheng, Ming-Ming}, booktitle={The IEEE International Conference on Computer Vision (ICCV)}, month={Oct}, year={2019}, }