ddddocr模型升级

wkx0 · Dec 21, 2021 · 6f8d5b1 · 6f8d5b1
1 parent 6b3d131
commit 6f8d5b1
Show file tree

Hide file tree

Showing 7 changed files with 540 additions and 447 deletions.
diff --git a/README.md b/README.md
@@ -17,17 +17,20 @@
 
 `pip install ddddocr`
 
-```
+```python
 import ddddocr
-
 ocr = ddddocr.DdddOcr()
-
 with open('test.png', 'rb') as f:
-
     img_bytes = f.read()
-
-res = ocr.classification(img_bytes)
-
+res = ocr.classification(img_bytes=img_bytes)
+print(res)
+```
+或者传入图片 base64 编码值（不包含图片头）
+```python
+import ddddocr
+ocr = ddddocr.DdddOcr()
+img_base64 = 'img_base64' # 示例
+res = ocr.classification(img_base64=img_base64)
 print(res)
 ```
 
@@ -44,4 +47,7 @@ print(res)
 
 |  参数名   | 默认值  | 说明  |
 |  ----  | ----  | ----  |
-| img  | 0 | bytes 图片的bytes格式 |
+| img_bytes  | None | bytes 图片的bytes格式 |
+| img_base64  | None | 图片的 base64 编码值（不包含图片头） |
+
+> 说明，当 `img_bytes` 和 `img_base64` 都存在时，优先使用 `img_bytes`
diff --git a/ddddocr/README.md b/ddddocr/README.md
@@ -2,11 +2,16 @@
 
 # 带带弟弟OCR通用验证码识别SDK免费开源版
 
+## 交流群（找对象，在苏州，dd群主）
+
+
+![qrcode.png](http://cdn.wenanzhe.com/Dingtalk_20211220105723.jpg!/scale/50)
+
 ## 环境要求
 
 `python >= 3.8`
 
-`Windows/Linux/Macox..`
+`Windows/Linux..`
 
 ## 调用方法
 
@@ -45,4 +50,6 @@ print(res)
 | img_bytes  | None | bytes 图片的bytes格式 |
 | img_base64  | None | 图片的 base64 编码值（不包含图片头） |
 
-> 说明，当 `img_bytes` 和 `img_base64` 都存在时，优先使用 `img_bytes`
+> 说明，当 `img_bytes` 和 `img_base64` 都存在时，优先使用 `img_bytes`
+
+> 如果使用GPU，需要自行安装cuda和cudnn，并在安装完ddddocr时执行 <br>`pip uninstall onnxrumtime`<br>然后手动执行<br>`pip install onnxruntime-gpu`
diff --git a/ddddocr/__init__.py b/ddddocr/__init__.py
diff --git a/ddddocr/common.onnx b/ddddocr/common.onnx
diff --git a/ddddocr/requirements.txt b/ddddocr/requirements.txt
@@ -1,4 +1,3 @@
-torchvision
 onnxruntime
 Pillow
 numpy
diff --git a/setup.py b/setup.py
@@ -14,7 +14,7 @@
 
 setup(
     name="ddddocr",
-    version="1.0.6",
+    version="1.2.0",
     author="sml2h3",
     description="带带弟弟OCR",
     long_description=long_description,
@@ -27,7 +27,7 @@
         "Operating System :: OS Independent",
     ],
     install_requires=['numpy', 'onnxruntime', 'Pillow'],
-    python_requires='>=3.8',
+    python_requires='<3.10',
     include_package_data=True,
     install_package_data=True,
 )
diff --git a/test.py b/test.py
@@ -2,7 +2,7 @@
 
 ocr = ddddocr.DdddOcr()
 
-with open('test.png', 'rb') as f:
+with open('test.jpg', 'rb') as f:
     img_bytes = f.read()
 
 res = ocr.classification(img_bytes)