27 lines
735 B
Python
27 lines
735 B
Python
from flask import Flask, request
|
|
from paddleclas import PaddleClas
|
|
|
|
from util.common_util import process_request
|
|
|
|
app = Flask(__name__)
|
|
CLAS = PaddleClas(model_name="text_image_orientation")
|
|
|
|
|
|
@app.route('/clas/orientation', methods=['POST'])
|
|
@process_request
|
|
def orientation():
|
|
"""
|
|
判断图片旋转角度，逆时针旋转该角度后为正。可能值["0", "90", "180", "270"]
|
|
:return: 最有可能的两个角度
|
|
"""
|
|
img_path = request.form.get('img_path')
|
|
clas_result = CLAS.predict(input_data=img_path)
|
|
clas_result = next(clas_result)[0]
|
|
if clas_result["scores"][0] < 0.5:
|
|
return ['0', '90']
|
|
return clas_result["label_names"]
|
|
|
|
|
|
if __name__ == '__main__':
|
|
app.run('0.0.0.0', 5002)
|