gushoubang 5 mēneši atpakaļ
vecāks
revīzija
edaf44c018
1 mainītis faili ar 16 papildinājumiem un 3 dzēšanām
  1. 16 3
      main.py

+ 16 - 3
main.py

@@ -30,7 +30,20 @@ def get_data():
     })
 
 # POST 上传单张图片
+@app.route('/api/upload_single_image', methods=['POST'])
+def upload_image():
+    # 获取上传的文件
+    img = request.files['img1']
+    # 保存文件
+    filePath = 'predict/' + str(uuid.uuid4()) + img.filename
 
+    img.save(filePath)
+    return jsonify({
+        "message": "上传成功",
+        "data":{
+            "img": filePath
+        }
+    })
 
 @app.route('/api/upload_single_image', methods=['POST'])
 def upload_single_image():
@@ -146,7 +159,7 @@ def save_image():
 
     return jsonify({
         "message": "保存成功",
-        "data": {
+        "data":{
             # "img1": img1Path,
             # "img2": img2Path,
             "img": 100,
@@ -160,10 +173,10 @@ def detect_image():
     data = request.get_json()
     img1 = data['img1']
     img2 = data['img2']
-    getImgPath = start(img1, img2)
+    getImgPath=start(img1,img2)
     return jsonify({
         "message": "分析成功",
-        "data": {
+        "data":{
             "img": host+'/predict/'+getImgPath
         }
     })