فهرست منبع

fix line_parser and ocr_model

zeke-chin 2 سال پیش
والد
کامیت
1d4ab4f087
1فایلهای تغییر یافته به همراه16 افزوده شده و 14 حذف شده
  1. 16 14
      server.py

+ 16 - 14
server.py

@@ -35,21 +35,23 @@ print(f'use gpu: {use_gpu}')
 
 # 初始化ocr模型和后处理模型
 # 分类
+# ocr = PaddleOCR(use_angle_cls=True,
+#                 # 方向
+#                 rec_model_dir="./idcard_rec_infer/",
+#                 det_model_dir="./idcard_det_infer/",
+#                 cls_model_dir="idcard_cls_infer",
+#                 # 识别
+#                 rec_algorithm='CRNN',
+#                 ocr_version='PP-OCRv2',
+#                 # 中文字典
+#                 rec_char_dict_path="./ppocr_keys_v1.txt", lang="ch",
+#                 use_gpu=use_gpu,
+#                 # 预训练-->效果不明显
+#                 # 网络不够大、不够深
+#                 # 数据集普遍较小,batch size普遍较小
+#                 warmup=True)
 ocr = PaddleOCR(use_angle_cls=True,
-                # 方向
-                rec_model_dir="./idcard_rec_infer/",
-                det_model_dir="./idcard_det_infer/",
-                cls_model_dir="idcard_cls_infer",
-                # 识别
-                rec_algorithm='CRNN',
-                ocr_version='PP-OCRv2',
-                # 中文字典
-                rec_char_dict_path="./ppocr_keys_v1.txt", lang="ch",
-                use_gpu=use_gpu,
-                # 预训练-->效果不明显
-                # 网络不够大、不够深
-                # 数据集普遍较小,batch size普遍较小
-                warmup=True)
+                use_gpu=use_gpu)
 
 
 # 初始化 角度检测器 对象