mirrors/mmocr
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmocr.git synced 2025-06-03 21:54:47 +08:00
Code Issues Projects Releases Wiki Activity

[Fix] Change mmcv.dump to mmengine.dump (#1540)

Browse Source 
- `dump` method was removed from MMCV 2.x. `Use mmengine.dump` instead.
This commit is contained in:
Janghoo Lee 2022-11-16 19:15:09 +09:00 committed by GitHub
parent 06a20fae71
commit b1a3b94508
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mmocr/apis/inferencers/mmocr_inferencer.py
View File

@ -4,6 +4,7 @@ from datetime import datetime
from typing import Dict, List, Optional, Tuple, Union from typing import Dict, List, Optional, Tuple, Union
import mmcv import mmcv
import mmengine
import numpy as np import numpy as np
from mmocr.registry import VISUALIZERS from mmocr.registry import VISUALIZERS
@ -210,7 +211,7 @@ class MMOCRInferencer(BaseMMOCRInferencer):
if print_result: if print_result:
print(results) print(results)
if pred_out_file != '': if pred_out_file != '':
mmcv.dump(results, pred_out_file) mmengine.dump(results, pred_out_file)
if imgs is None: if imgs is None:
return results return results
return results, imgs return results, imgs