mmdetection源码解读（九）

如何展示mmdet的检测结果&＃xff1a;

mmdetection/tools/bbox_show.py

import numpy as np
import mmcv
import cv2
import json
from pycocotools.coco import COCO
from mmcv import color_val
from mmdet.apis import show_result
from argparse import ArgumentParser
import osclass_names &＃61; [&＃39;category_1&＃39;, &＃39;category_2&＃39;, &＃39;category_3&＃39;]def det_bbox_show(args&＃61;None):""" 这个是用来展示检测结果的 """with open(args.json_file, &＃39;r&＃39;) as f:jsondata &＃61; json.load(f)results &＃61; {}for result in jsondata:result[&＃39;bbox&＃39;][2] &＃61; result[&＃39;bbox&＃39;][0] &＃43; result[&＃39;bbox&＃39;][2]result[&＃39;bbox&＃39;][3] &＃61; result[&＃39;bbox&＃39;][1] &＃43; result[&＃39;bbox&＃39;][3]result[&＃39;bbox&＃39;].append(result[&＃39;score&＃39;])if result[&＃39;image_id&＃39;] in results.keys():results[result[&＃39;image_id&＃39;]][result[&＃39;category_id&＃39;]-1] &＃61; np.concatenate((results[result[&＃39;image_id&＃39;]][result[&＃39;category_id&＃39;]-1], np.array(result[&＃39;bbox&＃39;], dtype&＃61;np.float32).reshape(-1,5)))else:results[result[&＃39;image_id&＃39;]] &＃61; []for i in range(len(class_names)):if result[&＃39;category_id&＃39;]-1 &＃61;&＃61; i:results[result[&＃39;image_id&＃39;]].append(np.array(result[&＃39;bbox&＃39;], dtype&＃61;np.float32).reshape(-1,5))else:results[result[&＃39;image_id&＃39;]].append(np.zeros(shape&＃61;(0,5),dtype&＃61;np.float32))for k, v in results.items():show_result(args.img_prex &＃43; &＃39;/&＃39; &＃43; k &＃43; &＃39;.jpg&＃39;, v, class_names, score_thr&＃61;0.7, show&＃61;False, out_file&＃61;args.outfile &＃43; &＃39;/{}.jpg&＃39;.format(k))def gt_bbox_show(args):"""这个是用来展示Ground Truth bbox的"""coco &＃61; COCO(args.json_file)img2ann &＃61; coco.imgToAnnsimgids &＃61; coco.getImgIds()for imgid in imgids:filename &＃61; args.img_prex &＃43; &＃39;/{}.jpg&＃39;.format(imgid)img &＃61; mmcv.imread(filename)bbox_color &＃61; color_val(&＃39;red&＃39;)text_color &＃61; color_val(&＃39;red&＃39;)anns &＃61; img2ann[imgid]for ann in anns:bbox_int &＃61; np.array(ann[&＃39;bbox&＃39;]).astype(np.int32)left_top &＃61; (bbox_int[0], bbox_int[1])right_bottom &＃61; (bbox_int[0] &＃43; bbox_int[2], bbox_int[1] &＃43; bbox_int[3])label &＃61; ann[&＃39;category_id&＃39;] - 1cv2.rectangle(img, left_top, right_bottom, bbox_color, thickness&＃61;1)label_text &＃61; class_names[label]cv2.putText(img, label_text, (bbox_int[0], bbox_int[1] - 2), cv2.FONT_HERSHEY_COMPLEX, 0.5, text_color)mmcv.imwrite(img, args.outfile &＃43; &＃39;/{}.jpg&＃39;.format(imgid))def main():parser &＃61; ArgumentParser(description&＃61;&＃39;COCO bbox show Tool&＃39;)parser.add_argument(&＃39;--gt-bbox-show&＃39;,action&＃61;&＃39;store_true&＃39;,help&＃61;&＃39;show gt bbox&＃39;)parser.add_argument(&＃39;--json_file&＃39;,type&＃61;str,help&＃61;&＃39;json file(results or ann file)&＃39;)parser.add_argument(&＃39;--img_prex&＃39;,type&＃61;str,help&＃61;&＃39;img dataset prex&＃39;)parser.add_argument(&＃39;--outfile&＃39;,type&＃61;str,help&＃61;&＃39;outfile&＃39;)args &＃61; parser.parse_args()if not os.path.exists(args.outfile):os.makedirs(args.outfile)if args.gt_bbox_show:gt_bbox_show(args)else:det_bbox_show(args)if __name__ &＃61;&＃61; &＃39;__main__&＃39;:import timetim &＃61; time.time()main()print("total time &＃61; {}".format(time.time() - tim))


这段程序前提是图片名称和图片id要保持一致&＃xff0c;像coco这种图片名前面补好多0的就不能用了&＃xff0c;需要debug一下。