[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:
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

View File

@ -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