mirrors
/
mmocr
mirror of https://github.com/open-mmlab/mmocr.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[Fix] Revert sync bn in inferencer (#1832)

pull/1840/head
Tong Gao 2023-04-04 10:14:51 +08:00 committed by GitHub
parent e6174b29fe
commit 16de16f8f8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mmocr/apis/inferencers/base_mmocr_inferencer.py
View File

@ -7,6 +7,7 @@ import mmengine
import numpy as np import numpy as np
from mmengine.dataset import Compose from mmengine.dataset import Compose
from mmengine.infer.infer import BaseInferencer, ModelType from mmengine.infer.infer import BaseInferencer, ModelType
from mmengine.model.utils import revert_sync_batchnorm
from mmengine.registry import init_default_scope from mmengine.registry import init_default_scope
from mmengine.structures import InstanceData from mmengine.structures import InstanceData
from rich.progress import track from rich.progress import track
@ -63,6 +64,7 @@ class BaseMMOCRInferencer(BaseInferencer):
init_default_scope(scope) init_default_scope(scope)
super().__init__( super().__init__(
model=model, weights=weights, device=device, scope=scope) model=model, weights=weights, device=device, scope=scope)
self.model = revert_sync_batchnorm(self.model)
def preprocess(self, inputs: InputsType, batch_size: int = 1, **kwargs): def preprocess(self, inputs: InputsType, batch_size: int = 1, **kwargs):
"""Process the inputs into a model-feedable format. """Process the inputs into a model-feedable format.