FCIS 经历记录

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

安装依赖linux

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)git

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

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

复制文件apache

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

可能出现问题:code

  • 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/

编译pythonip

cd python
sudo python setup.py install

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

./model/fcis_coco-0000.params

而后运行get

python ./fcis/demo.py
相关文章
相关标签/搜索