矫正扭曲和图片方向分类接口化

clas_api.py

@@ -0,0 +1,26 @@
+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)