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')

版权声明：本文为CSDN博主「小伟db」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/qq_35985044/article/details/114661869