diff --git a/ddddocr/README.md b/ddddocr/README.md
index 874851d..8248c6b 100644
--- a/ddddocr/README.md
+++ b/ddddocr/README.md
@@ -2,21 +2,27 @@
# 带带弟弟OCR通用验证码识别SDK免费开源版
-## 交流群(找对象,在苏州,dd群主)
+# 2021/12/24重大更新,ddddocr现在支持通用目标检测啦
+## 交流群(找对象,在苏州,dd群主)
+

## 环境要求
`python <= 3.9`
-`Windows/Linux..`
+`Windows/Linux/Macos..`
+
+暂时不支持Macbook M1(X),M1(X)用户需要自己编译onnxruntime才可以使用
## 调用方法
`pip install ddddocr`
+### 1、文字识别模式
+
```python
import ddddocr
ocr = ddddocr.DdddOcr()
@@ -34,17 +40,41 @@ res = ocr.classification(img_base64=img_base64)
print(res)
```
-### 参数说明
+### 2、目标检测模式
+```python
+import ddddocr
+det = ddddocr.DdddOcr(det=True)
-`DdddOcr 接受两个参数`
+with open('test.jpg', 'rb') as f:
+ img_bytes = f.read()
+
+res = det.detection(img_bytes)
+print(res)
+```
+
+### 3、参数说明
+
+`DdddOcr 接受三个参数`
| 参数名 | 默认值 | 说明 |
| ---- | ---- | ---- |
+| det | False | Bool 默认为识别文字模式,为True则开启目标检测模式 |
| use_gpu | False | Bool 是否使用gpu进行推理,如果该值为False则device_id不生效 |
| device_id | 0 | int cuda设备号,目前仅支持单张显卡 |
`classification`
+必须det参数为False后才可使用
+
+| 参数名 | 默认值 | 说明 |
+| ---- | ---- | ---- |
+| img_bytes | None | bytes 图片的bytes格式 |
+| img_base64 | None | 图片的 base64 编码值(不包含图片头) |
+
+`detection`
+
+必须det参数为False后才可使用
+
| 参数名 | 默认值 | 说明 |
| ---- | ---- | ---- |
| img_bytes | None | bytes 图片的bytes格式 |
@@ -52,4 +82,4 @@ print(res)
> 说明,当 `img_bytes` 和 `img_base64` 都存在时,优先使用 `img_bytes`
-> 如果使用GPU,需要自行安装cuda和cudnn,并在安装完ddddocr时执行
`pip uninstall onnxrumtime`
然后手动执行
`pip install onnxruntime-gpu`
\ No newline at end of file
+> 如果使用GPU,需要自行安装cuda和cudnn,并在安装完ddddocr时执行
`pip uninstall onnxrumtime`
然后手动执行
`pip install onnxruntime-gpu`
diff --git a/ddddocr/requirements.txt b/ddddocr/requirements.txt
index 68c5866..032f425 100644
--- a/ddddocr/requirements.txt
+++ b/ddddocr/requirements.txt
@@ -1,3 +1,4 @@
onnxruntime
Pillow
-numpy
\ No newline at end of file
+numpy
+opencv-python
\ No newline at end of file
diff --git a/setup.py b/setup.py
index c1d7789..5c20661 100644
--- a/setup.py
+++ b/setup.py
@@ -14,7 +14,7 @@ with open("README.md", "r", encoding="utf-8") as fh:
setup(
name="ddddocr",
- version="1.2.0",
+ version="1.2.1",
author="sml2h3",
description="带带弟弟OCR",
long_description=long_description,
@@ -26,7 +26,7 @@ setup(
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
],
- install_requires=['numpy', 'onnxruntime', 'Pillow'],
+ install_requires=['numpy', 'onnxruntime', 'Pillow', 'opencv-python'],
python_requires='<3.10',
include_package_data=True,
install_package_data=True,