论文code
https://github.com/zijundeng/R3Net
要求
Python 2.7
PyTorch 0.4.0
torchvision
numpy
Cython
pydensecrf
实现细节
- 安装环境,在此省略(可参考 here )
- 这里强调一下, pretrained ResNeXt model 最好是从作者提供的官网下载,作者试了从官网下载,反正调了好久没调通。
- 在运行过程中,出现了下面这个问题,仔细分析可能是论文作者是在双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'}))