This project link: https://github.com/lartpang/CMWNet.pytorch
This project provides an unofficial pytorch implementation of the RGB-D SOD method for 'Cross-Modal Weighting Network for RGB-D Salient Object Detection', ECCV 2020.
- Paper Link: https://arxiv.org/pdf/2007.04901.pdf
- Official Caffe Code: https://github.com/MathLee/CMWNet
torch>= 1.5.0 for AMP.torchvisionnumpytqdmopencv-pythonpillowpy_sod_metricsfor evaluating the results.
- Configure your datasets in
dataset_config.py. - Set hyperparameters in
main.py. python main.py- All results will be saved into the
ckpt_path(seemain.py).