Ubuntu16.04配置实例分割网络BlendMask

相关依赖版本说明

显卡：RTX2060
显卡驱动：440.82
Cuda：10.0， Cudnn7.6.5
Python3.7
gcc5.4.0，g++5.4.0
pytorch1.4.0，torchvision0.5.0
一、安装python环境

conda create -n BlendMask python=3.7
conda activate BlendMask
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple opencv-python
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.4.0 torchvision==0.5.0

二、安装Detectron2

python -m pip install detectron2 -f \
  https://dl.fbaipublicfiles.com/detectron2/wheels/cu100/torch1.4/index.html

三、安装AdelaiDet
参考：GitHub官网

git clone https://github.com/aim-uofa/AdelaiDet.git
cd AdelaiDet
python setup.py build develop

四、运行代码
1、下载权重
本人使用的权重是：R_101_dcni3_5x，下载地址
下载过后将其放在AdelaiDet文件夹下。
2、准备图片
准备两张图片，分别命名为input1.jpg,input2,jpg
3、运行命令

cd AdelaiDet
python demo/demo.py \
    --config-file configs/FCOS-Detection/R_50_1x.yaml \
    --input input1.jpg input2.jpg \
    --opts MODEL.WEIGHTS fcos_R_50_1x.pth

此时应该会显示检测结果，按下空格后会接着检测下一张图片，并显示