mirror of
https://github.com/open-mmlab/mmocr.git
synced 2025-06-03 21:54:47 +08:00
[Fix] Change mmcv.dump to mmengine.dump (#1540)
- `dump` method was removed from MMCV 2.x. `Use mmengine.dump` instead.
This commit is contained in:
parent
06a20fae71
commit
b1a3b94508
@ -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
|
||||
|
Loading…
x
Reference in New Issue
Block a user