|
@@ -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)
|
|
|
|
|
|
|
|
|
# 初始化 角度检测器 对象
|