save

main.py

@@ -30,17 +30,23 @@ def get_data():
     })
 
 # POST 上传单张图片
+
+
 @app.route('/api/upload_single_image', methods=['POST'])
-def upload_image():
+def upload_single_image():
     # 获取上传的文件
-    img = request.files['img1']
+    img = request.files['img']
+    # 文件夹不存在则创建
+    if not os.path.exists(uploadPath):
+        os.makedirs(uploadPath)
+
     # 保存文件
-    filePath = 'predict/' + str(uuid.uuid4()) + img.filename
+    filePath = uploadPath + str(uuid.uuid4()) + img.filename
 
     img.save(filePath)
     return jsonify({
         "message": "上传成功",
-        "data":{
+        "data": {
             "img": filePath
         }
     })
@@ -159,7 +165,7 @@ def save_image():
 
     return jsonify({
         "message": "保存成功",
-        "data":{
+        "data": {
             # "img1": img1Path,
             # "img2": img2Path,
             "img": 100,
@@ -173,10 +179,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
         }
     })