Merge pull request #273 from breezedeus/dev

fix: export_to_onnx
breezedeus · Aug 10, 2023 · 7c21696 · 7c21696
2 parents 777ba68 + 02791a0
commit 7c21696
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/Makefile b/Makefile
@@ -34,7 +34,7 @@ package:
 	rm -rf build
 	python setup.py sdist bdist_wheel
 
-VERSION = 2.2.3.1
+VERSION = 2.2.3.2
 upload:
 	python -m twine upload  dist/cnocr-$(VERSION)* --verbose
 

diff --git a/cnocr/__version__.py b/cnocr/__version__.py
@@ -17,4 +17,4 @@
 # specific language governing permissions and limitations
 # under the License.
 
-__version__ = '2.2.3.1'
+__version__ = '2.2.3.2'
diff --git a/cnocr/cli.py b/cnocr/cli.py
@@ -49,6 +49,7 @@
 from cnocr.dataset import OcrDataModule
 from cnocr.trainer import PlTrainer, resave_model
 from cnocr import CnOcr, gen_model
+from cnocr.recognizer import Recognizer
 
 _CONTEXT_SETTINGS = {"help_option_names": ['-h', '--help']}
 logger = set_logger(log_level=logging.INFO)
@@ -482,7 +483,7 @@ def resave_model_file(
 def export_to_onnx(model_name, output_model_fp, input_model_fp=None):
     import onnx
 
-    ocr = CnOcr(model_name, model_fp=input_model_fp)
+    ocr = Recognizer(model_name, model_fp=input_model_fp)
     model = ocr._model
 
     x = torch.randn(1, 1, 32, 280)