chore: add test files

Author: SWHL

demo.py

@@ -7,7 +7,7 @@
 import requests
 
 url = "http://localhost:9003/ocr"
-img_path = "../python/tests/test_files/ch_en_num.jpg"
+img_path = "tests/test_files/ch_en_num.jpg"
 
 # 方式一：使用base64编码传
 stime = time.time()
@@ -23,28 +23,28 @@
 
 print("-" * 40)
 
-# 方式二：使用文件上传方式
-stime = time.time()
-with open(img_path, "rb") as f:
-    file_dict = {"image_file": (img_path, f, "image/png")}
-    response = requests.post(url, files=file_dict)  # , timeout=60
-    print(response.json())
-
-etime = time.time() - stime
-print(f"用时:{etime:.3f}秒")
-print("-" * 40)
-
-# 方式三：控制是否使用检测、方向分类和识别这三部分的模型； 不使用检测模型:use_det=False
-stime = time.time()
-img_path = "../python/tests/test_files/test_without_det.jpg"
-
-with open(img_path, "rb") as f:
-    file_dict = {"image_file": (img_path, f, "image/png")}
-    # 添加控制参数
-    data = {"use_det": False, "use_cls": True, "use_rec": True}
-    response = requests.post(url, files=file_dict, data=data)  # , timeout=60
-    print(response.json())
-
-etime = time.time() - stime
-print(f"用时:{etime:.3f}秒")
-print("-" * 40)
+# # 方式二：使用文件上传方式
+# stime = time.time()
+# with open(img_path, "rb") as f:
+#     file_dict = {"image_file": (img_path, f, "image/png")}
+#     response = requests.post(url, files=file_dict)  # , timeout=60
+#     print(response.json())
+
+# etime = time.time() - stime
+# print(f"用时:{etime:.3f}秒")
+# print("-" * 40)
+
+# # 方式三：控制是否使用检测、方向分类和识别这三部分的模型； 不使用检测模型:use_det=False
+# stime = time.time()
+# img_path = "../python/tests/test_files/test_without_det.jpg"
+
+# with open(img_path, "rb") as f:
+#     file_dict = {"image_file": (img_path, f, "image/png")}
+#     # 添加控制参数
+#     data = {"use_det": False, "use_cls": True, "use_rec": True}
+#     response = requests.post(url, files=file_dict, data=data)  # , timeout=60
+#     print(response.json())
+
+# etime = time.time() - stime
+# print(f"用时:{etime:.3f}秒")
+# print("-" * 40)

rapidocr_api/main.py

@@ -13,18 +13,19 @@
 import uvicorn
 from fastapi import FastAPI, Form, UploadFile
 from PIL import Image
-from rapidocr import RapidOCR
-
-# if importlib.util.find_spec("rapidocr_onnxruntime"):
-#     from rapidocr_onnxruntime import RapidOCR
-# elif importlib.util.find_spec("rapidocr_paddle"):
-#     from rapidocr_paddle import RapidOCR
-# elif importlib.util.find_spec("rapidocr_openvino"):
-#     from rapidocr_openvino import RapidOCR
-# else:
-#     raise ImportError(
-#         "Please install one of [rapidocr_onnxruntime,rapidocr-paddle,rapidocr-openvino]"
-#     )
+
+# from rapidocr import RapidOCR
+
+if importlib.util.find_spec("rapidocr_onnxruntime"):
+    from rapidocr_onnxruntime import RapidOCR
+elif importlib.util.find_spec("rapidocr_paddle"):
+    from rapidocr_paddle import RapidOCR
+elif importlib.util.find_spec("rapidocr_openvino"):
+    from rapidocr_openvino import RapidOCR
+else:
+    raise ImportError(
+        "Please install one of [rapidocr_onnxruntime,rapidocr-paddle,rapidocr-openvino]"
+    )
 
 sys.path.append(str(Path(__file__).resolve().parent.parent))