复现R3Net: Recurrent Residual Refinement Network for Saliency Detection

论文code

https://github.com/zijundeng/R3Net

要求

Python 2.7
PyTorch 0.4.0
torchvision
numpy
Cython
pydensecrf

实现细节

1. 安装环境，在此省略（可参考 here )
2. 这里强调一下， pretrained ResNeXt model 最好是从作者提供的官网下载，作者试了从官网下载，反正调了好久没调通。
3. 在运行过程中，出现了下面这个问题，仔细分析可能是论文作者是在双GPU的情况下跑的，而我这里只有一个GPU，所以需要将GPU1转换成GPU0。
   出现问题：

RuntimeError: cuda runtime error (10) : invalid device ordinal at torch/csrc/cuda/Module.cpp:87

解决方法：加入map_location={‘cuda:1’:‘cuda:0’}参数项，可参考

load_state_dict(torch.load('params—xxxxx',map_location={'cuda:1':'cuda:0'}))