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