From daacbec8bbb95c2251b693dc45b14972ca6fdec3 Mon Sep 17 00:00:00 2001 From: wenaz Date: Fri, 7 Jan 2022 00:34:19 +0800 Subject: [PATCH] init --- Dockerfile | 17 ++++++++++ README.md | 70 +++++++++++++++++++++++++++++++++++++++ entrypoint.sh | 3 ++ ocr_server.py | 84 +++++++++++++++++++++++++++++++++++++++++++++++ requirements.txt | 2 ++ test.jpg | Bin 0 -> 2733 bytes test_api.py | 40 ++++++++++++++++++++++ 7 files changed, 216 insertions(+) create mode 100644 Dockerfile create mode 100644 entrypoint.sh create mode 100644 ocr_server.py create mode 100644 requirements.txt create mode 100644 test.jpg create mode 100644 test_api.py diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..4d2b306 --- /dev/null +++ b/Dockerfile @@ -0,0 +1,17 @@ +FROM python:3.8-buster + +RUN mkdir /app + +COPY ./*.txt ./*.py ./*.sh ./*.onnx /app/ + + +RUN cd /app \ + && python3 -m pip install --upgrade pip -i https://pypi.douban.com/simple/\ + && pip3 install --no-cache-dir -r requirements.txt --extra-index-url https://pypi.douban.com/simple/ \ + && rm -rf /tmp/* && rm -rf /root/.cache/* \ + && sed -i 's#http://deb.debian.org#http://mirrors.aliyun.com/#g' /etc/apt/sources.list\ + && apt-get --allow-releaseinfo-change update && apt install libgl1-mesa-glx -y + +WORKDIR /app + +ENTRYPOINT ["sh", "entrypoint.sh"] \ No newline at end of file diff --git a/README.md b/README.md index 6376bcb..93b0a51 100644 --- a/README.md +++ b/README.md @@ -1,2 +1,72 @@ # ocr_api_server 使用ddddocr的最简api搭建项目，支持docker + +**建议python版本3.7-3.9 64位** + +再有不好好看文档的我就不管了啊！！！ + +# 运行方式 + +## 最简单运行方式 + +```shell +# 安装依赖 +pip install -r requirements.txt -i https://pypi.douban.com/simple + +# 运行 可选参数如下 +# --port 9898 指定端口,默认为9898 +# --ocr 开启ocr模块 默认开启 +# --old 只有ocr模块开启的情况下生效 默认不开启 +# --det 开启目标检测模式 + +# 最简单运行方式，只开启ocr模块并以新模型计算 +python ocr_server.py --port 9898 --ocr + +# 开启ocr模块并使用旧模型计算 +python ocr_server.py --port 9898 --ocr -old + +# 只开启目标检测模块 +python ocr_server.py --port 9898 --det + +# 同时开启ocr模块以及目标检测模块 +python ocr_server.py --port 9898 --ocr --det + +# 同时开启ocr模块并使用旧模型计算以及目标检测模块 +python ocr_server.py --port 9898 --ocr --old --det + +``` + +## docker运行方式(目测只能在Linux下部署) + +```shell +git clone https://github.com/sml2h3/ocr_api_server.git +# docker怎么安装？百度吧 + +cd ocr_api_server + +# 修改entrypoint.sh中的参数，具体参数往上翻，默认9898端口，同时开启ocr模块以及目标检测模块 + +# 编译镜像 +docker build -t ocr_server:v1 . + +# 运行镜像 +docker run -p 9898:9898 -d ocr_server:v1 + +``` + +# 接口 + +**具体请看test_api.py文件** + +```python +# 1、测试是否启动成功，可以通过直接GET访问http://{host}:{port}/ping来测试，如果返回pong则启动成功 + +# 2、OCR请求 + +# resp = requests.post("http://{host}:{port}/ocr", files={'image': image_bytes}) + +# 3、目标检测请求 + +# resp = requests.post("http://{host}:{port}/det", files={'image': image_bytes}) + +``` diff --git a/entrypoint.sh b/entrypoint.sh new file mode 100644 index 0000000..8167127 --- /dev/null +++ b/entrypoint.sh @@ -0,0 +1,3 @@ +#!/bin/sh + +python3 ocr_server.py --port 9898 --ocr --det diff --git a/ocr_server.py b/ocr_server.py new file mode 100644 index 0000000..4bebfbb --- /dev/null +++ b/ocr_server.py @@ -0,0 +1,84 @@ +# encoding=utf-8 +import argparse +import json + +import ddddocr +from flask import Flask, request + +parser = argparse.ArgumentParser(description="使用ddddocr搭建的最简api服务") +parser.add_argument("-p", "--port", type=int, default=9898) +parser.add_argument("--ocr", action="store_true", help="开启ocr识别") +parser.add_argument("--old", action="store_true", help="OCR是否启动旧模型") +parser.add_argument("--det", action="store_true", help="开启目标检测") + +args = parser.parse_args() + +app = Flask(__name__) + + +class Server(object): + def __init__(self, ocr=True, det=False, old=False): + self.ocr_option = ocr + self.det_option = det + self.old_option = old + self.ocr = None + self.det = None + if self.ocr_option: + print("ocr模块开启") + if self.old_option: + print("使用OCR旧模型启动") + self.ocr = ddddocr.DdddOcr(old=True) + else: + print("使用OCR新模型启动，如需要使用旧模型，请额外添加参数 --old开启") + self.ocr = ddddocr.DdddOcr() + else: + print("ocr模块未开启，如需要使用，请使用参数 --ocr开启") + if self.det_option: + print("目标检测模块开启") + self.det = ddddocr.DdddOcr(det=True) + else: + print("目标检测模块未开启，如需要使用，请使用参数 --det开启") + + + def classification(self, img: bytes): + if self.ocr_option: + return self.ocr.classification(img) + else: + raise Exception("ocr模块未开启") + + def detection(self, img: bytes): + if self.det_option: + return self.det.detection(img) + else: + raise Exception("目标检测模块模块未开启") + + +server = Server(ocr=args.ocr, det=args.det, old=args.old) + + +@app.route('/ocr', methods=['POST']) +def ocr(): + try: + img = request.files.get('image').read() + r = server.classification(img) + return json.dumps({"status": "200", "result": str(r), "msg": ""}) + except Exception as e: + return json.dumps({"status": "500", "result": "", "msg": str(e)}) + +@app.route('/det', methods=['POST']) +def det(): + try: + img = request.files.get('image').read() + r = server.detection(img) + return json.dumps({"status": "200", "result": r, "msg": ""}) + except Exception as e: + return json.dumps({"status": "500", "result": "", "msg": str(e)}) + + +@app.route('/ping', methods=['GET']) +def ping(): + return "pong" + + +if __name__ == '__main__': + app.run(host="0.0.0.0", port=args.port) diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..a8661fa --- /dev/null +++ b/requirements.txt @@ -0,0 +1,2 @@ +ddddocr>=1.3.1 +flask \ No newline at end of file diff --git a/test.jpg b/test.jpg new file mode 100644 index 0000000000000000000000000000000000000000..d5894d2564b78304593c6cbdd1d58c44752d8c5f GIT binary patch literal 2733 zcmV;e3R3lnP)+&uA+7!+>xjk_l|5^QL!ztO^tg$^hN5`(i|z)HZ>Ya zB*$Ip-B~tIn?{Q!mkrldz`EIH^{@?*=ti?zwb_-h1l)qhmdyd8;kZA(FgrW*-s7>T zx%-Ap-p>2ZcV<5C^Zm|uhJic)SrS*STq#T9#KZ(s?%TIdmc;&h_n3NYh4l{}JRr|k z0yju289Nq|36MduCb2ZSh!O*%iJ` z5aHiq4I;(kMO%1nN3kr4p|*O00zM;NsHuU}z$JnRmWU}Lg%}OpZX+d+6hG&+um3hV zyZ|jWH#d`+<2R?d!!wK4;qgLih}VgaUY8}YyS;{#jfb|#l9<2sIii2z!UclSp7*xI zEe~CbGmA!w!Vd95jIca&?(;{;CT(W%+@VpR-^vt@h?<@!PI>noSrXT8UM5fZBeG^m zQ7*$QxPUcU_|TC&mc>p^Ve~Ypl7bftUWKrF*4(c zsDtPrQrIKb^UOjXAtH3lMowW3zZ%ey4&S2xB5KIES$blM-EN2fTdAzBX{@h%0|0L@IV2pK15JxSS4OFeizbpeTw00II6aek3OQLec9n=yO{)y^$odTuujK zswfKG0rSx+t7`zj3}nP0Eo2&cbHMIQGKB_(7Bim&nKc8jH0Pr?ECU2*WG&yCQItJz zdGSU7kmse`x<65SFy2ZJCPh(FvxMS>J9NVxyVu(bMDprt9}Gm&8GU0XHD7Na-NBo~p>T@Z>oXZs^j^_(=4oYqpxBmS z%G%`*<}I3t>(n+xdjw(%02=G-NRv|Q7(JLr2-deAPo^ZM@y%#k1A+mVmrSSyP1qs&17fz~XwraoP=&R!q_L;xUYO^tsZMVWR@ zNKP^(aX1v#a6z_8^BglVrtal2bio*UHcG&M^wW`jvVW~H2Z`%V0P z{8os5y>a8l?Hd;9HU2<#-S&_0gVPk(mvsPu^Z&GN9paA5MeO>~^74(0z9$eqcI5nI z+i7?@)q8D2#fvrkr>d;3A#36+SHs~@I4z^>Rcz^yC76P2$L{s^0$rp4;E+;r;r95$ z;S5rP>;+ZOXBE6;E!oGFPZ&CSiUsza9M{!-+r4pWXm z#!AR9jIvNJNtETWCf^2SC#PDJWN+u)TeR{-FpE%^W<0N_`hFRcPNFXTm?)s8vo z6O45RZi}tj>(`ixClH?VF#VsGn*iS8Ls<6WW%SS3P}j+GlkvX0qa6yi!4%4Q%Zej| zlj|cdd_5g)@xcJK%Icc8N^O~vn1&d^lmiwP7Sehr+6G%%TJZUc^Evi{_^TyPi2(4* zrk61Q8;j5PSJ0aLu_NsO=K)}CNvAJZOz;XrI=~2Hq$8ZFXaBqQqa?=b?H@El72~9u zkTurV;c7VN_^nV`)fy7rpZ|H5=4hV!sdd+@1$52oToCf241-j)NvUo4Ky@+Ae^Imz z07Mw``r|1jGx_mY)ECU5_X#2-1g6o@HS+v)$vXN~fBz6ZVB${dT}}t?A0RO!<3&|f z19D-^N`u54>ih7Ifx=|_hD9r%cL2cZTn!U%{-%ol=?R3r{&4^y+B3qe@vblHm~;se zCl8a}9RUDeFsFZ@7eL6)UO+3%msV9)7dO_|m0DAokL2}_;{(GTyL-Y1cGg&5N0>F!{9c<`E~n$OV_pUqJ@|-VU{;iEpVqB&IUP)% zNjNxPTBX^XnH%ai*4I58?Ia-#E9sDjmDPVc%^Ng%lw>mF1v+N&-Pj9C-#+LaTbKg? z=|4pJ`uYIi{SOmNfAU$wOV3G=7&|lxrMHUVlIL9pF?GesGsJvXGd2Fc5 zlp7B8x|~%NyN)YL4Zv7zs0(50?r5hw0)WxbwafkYZ~I4n=WhpSR{<17$rNQpncksH z9;Gq~2aTi4>DaLED~(OaRDJBgo-Foe=OI2l>AY@qWimRb4Mj6+HV zm_9@XfS|7)6C(5!AumccZYa$B8APT%FMN`sHyAvF*Er8|IURh^x!MQk+OD^@w$hbl zWo3JJt}qWPvUSbo!(X4?yJmAjCUx*n4IloYiqY4sWMu#T{Ra*lF#k%=CZAx^qW#|; z;>(hJnV4W5blfr)k^I{AZ#;o{ihzul0 zWzilX+C#Ro0>E9~#NE4hTU%TIGdG`K0r0E>k1J5k?!>xI zX~tA$;||oOL391ew@w}e_7C)uLtz8`{}C}cy!dLETzh!TvH6i0H~)`eW5Jj;2e@6O zc_KV=q-K;!nc1wEM@r8c)(3rVi>(}F!#j@)o)wt;KzNIYF7d`1-UPyXLJwl!$&wiB zir-iSNHJORx7v(L`c{{D{p4=Afh>vm$K!?1G+xF?;oUR53yIHoqbfP1W{ScSz)OQ! zSmNzH{#F~JL*q5s_TdiB~@nJN}`^*0YHuokJ2^!s200000NkvXXu0mjf^v^s& literal 0 HcmV?d00001 diff --git a/test_api.py b/test_api.py new file mode 100644 index 0000000..bc8279d --- /dev/null +++ b/test_api.py @@ -0,0 +1,40 @@ +#!/usr/bin/python3.6 +# -*- coding: utf-8 -*- +# +# Copyright (C) 2021 # +# @Time : 2022/1/6 23:28 +# @Author : sml2h3 +# @Email : sml2h3@gmail.com +# @File : test_api.py +# @Software: PyCharm +import requests + +# ******************OCR识别部分开始****************** +# 目标检测就把ocr改成det,其他相同 +api_url = "http://10.0.20.198:9898/ocr" + +# 方式一 +# file = open(r'test.jpg', 'rb').read() +# +# resp = requests.post(api_url, files={'image': file}) +# print(resp.text) + +# 方式二 + +# 获取验证码图片 +# headers = { +# "Content-Type": "application/x-www-form-urlencoded; charset=UTF-8", +# "User-Agent": "Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/86.0.4195.1 Safari/537.36" +# } +# resp = requests.get('https://data.gdcic.net/Dop/CheckCode.aspx?codemark=408.15173910730016', headers=headers, verify=False) +# captcha_img = resp.content +# +# 识别 +# resp = requests.post(api_url, files={'image': captcha_img}) +# print('验证码结果', resp.text) +# +# # 保存验证码图片以供验证 +# with open('captcha.jpg', 'wb') as f: +# f.write(captcha_img) + +# ******************OCR识别部分开始******************