'TextRecogDataPreprocessor' is not in the model registry.

[Subhajyotidas]

I have trained text recognition model on Japanese dataset.
I did not change much architecture and just followed the toy dataset method by just replacing the dataset path and character dictionary path in the config files. Every other setting is same.
The training was successful, Now I am trying to do inference with the traind model, but the following problem kept showing up.

04/07 09:43:04 - mmengine - WARNING - The "model" registry in mmocr did not set import location. Fallback to call `mmocr.utils.register_all_modules` instead.
Traceback (most recent call last):
  File "D:\Anaconda\envs\open-mmlab\lib\site-packages\mmengine\registry\build_functions.py", line 121, in build_from_cfg
    obj = obj_cls(**args)  # type: ignore
  File "D:\nn_practice\mmocr\mmocr\mmocr\models\textrecog\recognizers\encoder_decoder_recognizer.py", line 43, in __init__
    super().__init__(
  File "D:\nn_practice\mmocr\mmocr\mmocr\models\textrecog\recognizers\base.py", line 26, in __init__
    super().__init__(
  File "D:\Anaconda\envs\open-mmlab\lib\site-packages\mmengine\model\base_model\base_model.py", line 78, in __init__
    self.data_preprocessor = MODELS.build(data_preprocessor)
  File "D:\Anaconda\envs\open-mmlab\lib\site-packages\mmengine\registry\registry.py", line 521, in build
    return self.build_func(cfg, *args, **kwargs, registry=self)
  File "D:\Anaconda\envs\open-mmlab\lib\site-packages\mmengine\registry\build_functions.py", line 240, in build_model_from_cfg
    return build_from_cfg(cfg, registry, default_args)
  File "D:\Anaconda\envs\open-mmlab\lib\site-packages\mmengine\registry\build_functions.py", line 100, in build_from_cfg
    raise KeyError(
KeyError: 'TextRecogDataPreprocessor is not in the model registry. Please check whether the value of `TextRecogDataPreprocessor` is correct or it was registered as expected. More details can be found at https://mmengine.readthedocs.io/en/latest/advanced_tutorials/config.html#import-the-custom-module'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "test_inference.py", line 9, in <module>
    ocr = MMOCRInferencer(rec_config =config_path, rec_ckpt=weight_path)
  File "D:\nn_practice\mmocr\mmocr\mmocr\apis\inferencers\mmocr_inferencer.py", line 41, in __init__
    self.textrec_inferencer = TextRecInferencer(
  File "D:\nn_practice\mmocr\mmocr\mmocr\apis\inferencers\base_mmocr_inferencer.py", line 59, in __init__
    super().__init__(config=config, ckpt=ckpt, device=device, **kwargs)
  File "D:\nn_practice\mmocr\mmocr\mmocr\apis\inferencers\base_inferencer.py", line 64, in __init__
    self._init_model(cfg, ckpt, device)
  File "D:\nn_practice\mmocr\mmocr\mmocr\apis\inferencers\base_inferencer.py", line 73, in _init_model
    model = MODELS.build(cfg.model)
  File "D:\Anaconda\envs\open-mmlab\lib\site-packages\mmengine\registry\registry.py", line 521, in build
    return self.build_func(cfg, *args, **kwargs, registry=self)
  File "D:\Anaconda\envs\open-mmlab\lib\site-packages\mmengine\registry\build_functions.py", line 240, in build_model_from_cfg
    return build_from_cfg(cfg, registry, default_args)
  File "D:\Anaconda\envs\open-mmlab\lib\site-packages\mmengine\registry\build_functions.py", line 135, in build_from_cfg
    raise type(e)(
KeyError: "class `SARNet` in mmocr/models/textrecog/recognizers/sar.py: 'TextRecogDataPreprocessor is not in the model registry. Please check whether the value of `TextRecogDataPreprocessor` is correct or it was registered as expected. More details can be found at https://mmengine.readthedocs.io/en/latest/advanced_tutorials/config.html#import-the-custom-module'"

Any idea how to deal with this problem?

[gaotongxiao]

Looks like a bug prior to the complete refactor of MMOCRInferencer. You can upgrade your code to the latest 1.0.0 release and try again.

[gaotongxiao]

Just upgrade your mmengine to 0.7.1: https://mmocr.readthedocs.io/en/dev-1.x/get_started/install.html#dependency-on-mmengine-mmcv-mmdetection

[Subhajyotidas]

I upgraded it, restarted the environment.
Then ran the inference code.
Same error, except the warning at the beginning.

[gaotongxiao]

Does this verification work in your environment? https://mmocr.readthedocs.io/en/dev-1.x/get_started/install.html#verify-the-installation

If it works, you can report the bug here: https://github.com/open-mmlab/mmocr/issues/new?assignees=&labels=bug&template=1-bug-report.yml&title=%5BBug%5D+

[Subhajyotidas]

It did not work.
Should I clone the github again?

[gaotongxiao]

Just re-install MMOCR from source in your environment and try running the demo again:

# in the root directory of mmocr, run:
pip install -v -e .