Awesome
A Multi-Task Collaborative Network for Light Field Salient Object Detection
Training stage
- Prepare the training data and change the data path
- Download the initial model - hqpw to the weights folder
- Train the model: python3 run_train.py --mode train
Testing stage
- Download the pretrained model- rrg7
- Download a example of testing data -9249
- 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}, }