5
0
mirror of https://github.com/sml2h3/ddddocr.git synced 2025-05-04 14:53:16 +08:00

新增目标检测功能

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

View File

@ -2,21 +2,27 @@
# 带带弟弟OCR通用验证码识别SDK免费开源版 # 带带弟弟OCR通用验证码识别SDK免费开源版
## 交流群找对象在苏州dd群主 # 2021/12/24重大更新ddddocr现在支持通用目标检测啦
## 交流群找对象在苏州dd群主
![qrcode.png](http://cdn.wenanzhe.com/Dingtalk_20211220105723.jpg!/scale/50) ![qrcode.png](http://cdn.wenanzhe.com/Dingtalk_20211220105723.jpg!/scale/50)
## 环境要求 ## 环境要求
`python <= 3.9` `python <= 3.9`
`Windows/Linux..` `Windows/Linux/Macos..`
暂时不支持Macbook M1(X)M1(X)用户需要自己编译onnxruntime才可以使用
## 调用方法 ## 调用方法
`pip install ddddocr` `pip install ddddocr`
### 1、文字识别模式
```python ```python
import ddddocr import ddddocr
ocr = ddddocr.DdddOcr() ocr = ddddocr.DdddOcr()
@ -34,17 +40,41 @@ res = ocr.classification(img_base64=img_base64)
print(res) 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不生效 | | use_gpu | False | Bool 是否使用gpu进行推理如果该值为False则device_id不生效 |
| device_id | 0 | int cuda设备号目前仅支持单张显卡 | | device_id | 0 | int cuda设备号目前仅支持单张显卡 |
`classification` `classification`
必须det参数为False后才可使用
| 参数名 | 默认值 | 说明 |
| ---- | ---- | ---- |
| img_bytes | None | bytes 图片的bytes格式 |
| img_base64 | None | 图片的 base64 编码值(不包含图片头) |
`detection`
必须det参数为False后才可使用
| 参数名 | 默认值 | 说明 | | 参数名 | 默认值 | 说明 |
| ---- | ---- | ---- | | ---- | ---- | ---- |
| img_bytes | None | bytes 图片的bytes格式 | | img_bytes | None | bytes 图片的bytes格式 |

View File

@ -1,3 +1,4 @@
onnxruntime onnxruntime
Pillow Pillow
numpy numpy
opencv-python

View File

@ -14,7 +14,7 @@ with open("README.md", "r", encoding="utf-8") as fh:
setup( setup(
name="ddddocr", name="ddddocr",
version="1.2.0", version="1.2.1",
author="sml2h3", author="sml2h3",
description="带带弟弟OCR", description="带带弟弟OCR",
long_description=long_description, long_description=long_description,
@ -26,7 +26,7 @@ setup(
"License :: OSI Approved :: MIT License", "License :: OSI Approved :: MIT License",
"Operating System :: OS Independent", "Operating System :: OS Independent",
], ],
install_requires=['numpy', 'onnxruntime', 'Pillow'], install_requires=['numpy', 'onnxruntime', 'Pillow', 'opencv-python'],
python_requires='<3.10', python_requires='<3.10',
include_package_data=True, include_package_data=True,
install_package_data=True, install_package_data=True,