In [1]:

    

from mmdet.apis import init_detector, inference_detector, show_result_pyplot
import mmcv


    

In [2]:

    

config_file = '../configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py'
# download the checkpoint from model zoo and put it in `checkpoints/`
checkpoint_file = '../checkpoints/faster_rcnn_r50_fpn_1x_20181010-3d1b3351.pth'


    

In [3]:

    

# build the model from a config file and a checkpoint file
model = init_detector(config_file, checkpoint_file, device='cuda:0')


    

In [4]:

    

# test a single image
img = 'demo.jpg'
result = inference_detector(model, img)


    

In [5]:

    

# show the results
show_result_pyplot(model, img, result)