diff --git a/mmocr/apis/inferencers/mmocr_inferencer.py b/mmocr/apis/inferencers/mmocr_inferencer.py index 5a33e8a3..32714092 100644 --- a/mmocr/apis/inferencers/mmocr_inferencer.py +++ b/mmocr/apis/inferencers/mmocr_inferencer.py @@ -4,6 +4,7 @@ from datetime import datetime from typing import Dict, List, Optional, Tuple, Union import mmcv +import mmengine import numpy as np from mmocr.registry import VISUALIZERS @@ -210,7 +211,7 @@ class MMOCRInferencer(BaseMMOCRInferencer): if print_result: print(results) if pred_out_file != '': - mmcv.dump(results, pred_out_file) + mmengine.dump(results, pred_out_file) if imgs is None: return results return results, imgs