[PyTorch] Facebook Research - Mask R-CNN Benchmark 的安装与测试

Github项目链接：https://github.com/facebookresearch/maskrcnn-benchmark

maskrcnn_benchmark 安装步骤：

• 安装Anaconda3，创建虚拟环境。

conda activate maskrcnn

conda create -n maskrcnn python=3

conda activate maskrcnn

• 在虚拟环境中安装依赖包。

conda install ipython

pip install ninja yacs cython matplotlib tqdm opencv-python

• 安装PyTorch。

conda install -c pytorch pytorch-nightly torchvision cudatoolkit=9.0

• 选择安装目录。

mkdir maskrcnn

export INSTALL_DIR=$PWD

cd $INSTALL_DIR

• 卸载torchvision 0.3.0，安装torchvision 0.2.2

pip uninstall torchvision

pip install torchvision==0.2.2

• 安装pycocotools。

git clone https://github.com/cocodataset/cocoapi.git

cd cocoapi/PythonAPI

python setup.py build_ext install

• 安装apex。

cd $INSTALL_DIR

git clone https://github.com/NVIDIA/apex.git

cd apex

python setup.py install --cuda_ext --cpp_ext

• 安装maskrcnn-benchmark。

cd $INSTALL_DIR

git clone https://github.com/facebookresearch/maskrcnn-benchmark.git

cd maskrcnn-benchmark

python setup.py build develop

unset INSTALL_DIR

---

maskrcnn-benchmark 测试：

• 进入maskrcnn-benchmark安装目录下的demo文件夹。

conda activate maskrcnn

cd maskrcnn/maskrcnn-benchmark/demo/

• 在demo目录下新建demo.py文件。

from maskrcnn_benchmark.config import cfg
from predictor import COCODemo
import matplotlib.pylab as pylab
import matplotlib.pyplot as plt
import cv2


pylab.rcParams['figure.figsize'] = 20, 12


def show_image(image):
    plt.imshow(image[:, :, [2, 1, 0]])
    plt.axis('off')
    plt.show()


config_file = '../configs/caffe2/e2e_mask_rcnn_X-152-32x8d-FPN-IN5k_1.44x_caffe2.yaml'
cfg.merge_from_file(config_file)

coco_demo = COCODemo(
    cfg,
    confidence_threshold=0.7,
    min_image_size=800
)

img = cv2.imread('path-to-coco2014/val2014/COCO_val2014_000000000772.jpg')
show_image(img)

predictions = coco_demo.run_on_opencv_image(img)
show_image(predictions)