From 316afec0a04da07442ce0c7202df8eb8f9d2047f Mon Sep 17 00:00:00 2001 From: gaotongxiao Date: Thu, 20 Jul 2023 17:26:35 +0800 Subject: [PATCH] [Fix] Return bboxes in MMOCRInferencer --- mmocr/apis/inferencers/mmocr_inferencer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/mmocr/apis/inferencers/mmocr_inferencer.py b/mmocr/apis/inferencers/mmocr_inferencer.py index c531be35c..ae4fdc52d 100644 --- a/mmocr/apis/inferencers/mmocr_inferencer.py +++ b/mmocr/apis/inferencers/mmocr_inferencer.py @@ -381,6 +381,7 @@ def postprocess(self, pred_results[i].update( dict( det_polygons=det_dict_res['polygons'], + det_bboxes=det_dict_res['bboxes'], det_scores=det_dict_res['scores'])) if 'kie' in self.mode: for i, kie_pred in enumerate(preds['kie']):