FCIS 经历记录

FCIS https://github.com/msracver/FCIS COCO2016分割比赛冠军

安装依赖

pip install Cython
pip install opencv-python==3.2.0.6
pip install easydict==1.6
pip install hickle

下载mxnet 文中明确说过,测试的时候使用[MXNet@(commit 62ecb60)](https://github.com/apache/incubator-mxnet/tree/62ecb60 "MXNet@(commit 62ecb60),训练的时候使用[MXNet@(commit 998378a)](https://github.com/apache/incubator-mxnet/tree/998378a "MXNet@(commit 998378a)

我们这里直接使用[MXNet@(commit 998378a)](https://github.com/apache/incubator-mxnet/tree/998378a "MXNet@(commit 998378a),

git clone --recursive https://github.com/dmlc/mxnet.git
git checkout 998378a
git submodule update  --init --recursive

复制文件

cp -r $(FCIS_ROOT)/fcis/operator_cxx/channel_operator* $(MXNET_ROOT)/src/operator/contrib/

编译

cd ${MXNET_ROOT}
make -j $(nproc) USE_OPENCV=1 USE_BLAS=openblas USE_CUDA=1 USE_CUDA_PATH=/usr/local/cuda USE_CUDNN=1

可能出现问题:

  • CUDA_R_32I 没有定义, 打开mshadow下的mshadow/base.h, 把CUDA_R_32I改成CUDA_R_32F
  • -lcuda找不到,找到libcuda.so在系统中的路径,然后添加到make/config.mk中类似ADD_LDFLAGS += /usr/lib/x86_64-linux-gnu/

编译python

cd python
sudo python setup.py install

下载已有模型百度云密码tmd4, 将模型放在model文件夹下,类似

./model/fcis_coco-0000.params

然后运行

python ./fcis/demo.py

猜你喜欢

转载自my.oschina.net/u/1046919/blog/1835430