新增目标检测功能

This commit is contained in:
wenaz
2021-12-24 15:50:28 +08:00
parent 6f49b3d95a
commit af1d76cd53
3 changed files with 39 additions and 8 deletions
+35 -5
View File
@@ -2,21 +2,27 @@
# 带带弟弟OCR通用验证码识别SDK免费开源版
## 交流群(找对象,在苏州,dd群主)
# 2021/12/24重大更新,ddddocr现在支持通用目标检测啦
## 交流群(找对象,在苏州,dd群主)
![qrcode.png](http://cdn.wenanzhe.com/Dingtalk_20211220105723.jpg!/scale/50)
## 环境要求
`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时执行 <br>`pip uninstall onnxrumtime`<br>然后手动执行<br>`pip install onnxruntime-gpu`
> 如果使用GPU,需要自行安装cuda和cudnn,并在安装完ddddocr时执行 <br>`pip uninstall onnxrumtime`<br>然后手动执行<br>`pip install onnxruntime-gpu`
+2 -1
View File
@@ -1,3 +1,4 @@
onnxruntime
Pillow
numpy
numpy
opencv-python