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