mask rcnn demo中test的ipynb版

话不多说上代码

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

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

config_file="../configs/caffe2/e2e_mask_rcnn_R_50_FPN_1x_caffe2.yaml"
cfg.merge_from_file(config_file)

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

def imshow(img):
    plt.imshow(img[:, :, [2, 1, 0]])
    plt.axis("off")
    plt.show()

img=cv2.imread('6.jpg')
imshow(img)

composite = coco_demo.run_on_opencv_image(img)
# rimg = cv2.imread('1.jpg')
imshow(composite)

运行环境 maskrcnn——benchmark官方要求环境配置好的情况下，在deom文件夹中运行；jupyter notebook 中运行