调整图片传入方式

main.py

@@ -1,6 +1,9 @@
 from flask import Flask, request, jsonify
 import uuid
 from change_detect import start
+import requests
+from PIL import Image
+from io import BytesIO
 
 
 app = Flask(__name__)
@@ -37,21 +40,60 @@ def upload_image():
             "img2": filePath2
         }
     })
-
-# POST 分析图片
-@app.route('/api/detect_image', methods=['POST'])
-def detect_image():
+# POST 通过url保存图片到本地
+@app.route('/api/save_image', methods=['POST'])
+def save_image():
     # 获取图片路径
     data = request.get_json()
     img1 = data['img1']
     img2 = data['img2']
+    # 保存文件
+    filePath1 = 'predict/' + str(uuid.uuid4()) + '.png'
+    filePath2 = 'predict/' + str(uuid.uuid4()) + '.png'
+
+    response1 = requests.get('http://192.168.60.63'+img1)
+    if response1.status_code != 200:
+        return jsonify({
+            "message": "图片1不存在",
+            "data":{
+                "img1": img1
+            }
+        })
+    img1 = Image.open(BytesIO(response1.content))
+
+    response2 = requests.get('http://192.168.60.63'+img2)
+    if response2.status_code != 200:
+        return jsonify({
+            "message": "图片2不存在",
+            "data":{
+                "img2": img2
+            }
+        })
+    img2 = Image.open(BytesIO(response2.content))
+
     getImgPath=start(img1,img2)
+
     return jsonify({
-        "message": "分析成功",
+        "message": "保存成功",
         "data":{
-            "img": getImgPath
+            "imgRes":getImgPath
         }
     })
 
+# # POST 分析图片
+# @app.route('/api/detect_image', methods=['POST'])
+# def detect_image():
+#     # 获取图片路径
+#     data = request.get_json()
+#     img1 = data['img1']
+#     img2 = data['img2']
+#     getImgPath=start(img1,img2)
+#     return jsonify({
+#         "message": "分析成功",
+#         "data":{
+#             "img": getImgPath
+#         }
+#     })
+
 if __name__ == '__main__':
     app.run(debug=True,port=4100)

predict/b8ed3bcb-554c-4e97-9071-7cded8ce950d.png

@@ -0,0 +1 @@
+{"code": "not_found", "message": "File not found or signature is invalid", "status": 404}

predict/cdf3738b-1f3a-4ab0-8ac7-16bb856bde87.png

@@ -0,0 +1 @@
+{"code": "not_found", "message": "File not found or signature is invalid", "status": 404}