mmdetection测试单张图片

from mmdet.apis import init_detector
from mmdet.apis import inference_detector
#from mmdet.apis import show_result
 
# 模型配置文件
config_file = './configs/retinanet/retinanet_r101_fpn_1x_coco.py'

# 预训练模型文件
checkpoint_file = '/res101_retina_l1_coco/epoch_12.pth'

 
# 通过模型配置文件与预训练文件构建模型
model = init_detector(config_file, checkpoint_file, device='cuda:0')
 
# 测试单张图片并进行展示
img = '/coco/images/test2017/xxx.jpg'
# img = './nms_test.jpg'
result = inference_detector(model, img)
model.show_result(img, result, model.CLASSES, out_file='xxx.jpg')