[Fix] Return bboxes in MMOCRInferencer

open-mmlab · Jul 20, 2023 · 316afec · 316afec
1 parent 1dcd6fa
commit 316afec
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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']):