Skip to content

可上传图片识别 #9

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
moeshin wants to merge 3 commits into
base: master
Choose a base branch
from
Open

可上传图片识别 #9

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
1d3ab64
可上传图片识别
moeshin Apr 2, 2023
97ec2b3
修复命令行参数
moeshin Apr 2, 2023
33b5efe
请求参数名的格式统一为大驼峰
moeshin Apr 4, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
46 changes: 30 additions & 16 deletions app.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,8 @@
app = Flask(__name__)
app.config['JSON_AS_ASCII'] = False

enable_image_path = True


# 失败的返回
def jsonFail(message):
Expand Down Expand Up @@ -71,19 +73,19 @@ def ocrResultSort(ocr_result):


# ocr解析
def ocrProcess(imgPath, language):
def ocrProcess(img, language):
if language == "JAP":
result = japOcr.ocr(imgPath, cls=False)
result = japOcr.ocr(img, cls=False)
elif language == "ENG":
result = engOcr.ocr(imgPath, cls=False)
result = engOcr.ocr(img, cls=False)
elif language == "KOR":
result = korOcr.ocr(imgPath, cls=False)
result = korOcr.ocr(img, cls=False)
elif language == "RU":
result = ruOcr.ocr(imgPath, cls=False)
result = ruOcr.ocr(img, cls=False)
elif language == "ZH":
result = zhOcr.ocr(imgPath, cls=False)
result = zhOcr.ocr(img, cls=False)
else:
result = japOcr.ocr(imgPath, cls=False)
result = japOcr.ocr(img, cls=False)

try:
result = ocrResultSort(result)
Expand Down Expand Up @@ -122,14 +124,22 @@ def handle_request():
})

try:
post_data = request.get_data()
post_data = json.loads(post_data.decode("utf-8"))

languageList = ["JAP", "ENG", "KOR", "RU", "ZH"]
if post_data["Language"] not in languageList:
return jsonFail("Language {} doesn't exist".format(post_data["Language"]))

res = ocrProcess(post_data["ImagePath"], post_data["Language"])
if request.mimetype == "multipart/form-data":
post_data = request.form
image = request.files["Image"].stream.read()
else:
if not enable_image_path:
return jsonFail("Disable image path")
post_data = request.get_data()
post_data = json.loads(post_data.decode("utf-8"))
image = post_data["ImagePath"]
language = post_data["Language"]

language_list = ["JAP", "ENG", "KOR", "RU", "ZH"]
if language not in language_list:
return jsonFail("Language {} doesn't exist".format(language))

res = ocrProcess(image, language)
return jsonSuccess(res)

except Exception as err:
Expand All @@ -145,12 +155,16 @@ def handle_request():
parser.add_argument("-h", "--host", type=str, default=host, help="监听的主机。默认：\"%s\"" % host)
parser.add_argument("-p", "--port", type=int, default=port, help="监听的端口。默认：%d" % port)
parser.add_argument("-P", "--path", type=str, default=path, help="监听的路径。默认：\"%s\"" % path)
parser.add_argument("-d", "--disable-image-path", action="store_true", help="禁止图片路径。")
parser.add_argument('--help', action='help', help='打印帮助。')
args = parser.parse_args()

host = args.host
port = args.port
path = args.path
print("监听：http://%s:%d%s" % (host, port, path))
enable_image_path = not args.disable_image_path

print('是否允许图片路径：%s' % enable_image_path)
print("接口：http://%s:%d%s" % (host, port, path))
app.add_url_rule(path, view_func=handle_request, methods=["POST", "HEAD"])
app.run(debug=False, host=host, port=port, threaded=False)