[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=
 
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.

• 安装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)