Update app.py

app.py

@@ -13,53 +13,60 @@ def get_token(client_id, client_secret):
     resultJson = response.json()
     return resultJson['access_token']
 
-def face_merge(image_template,image_target):
+def face_merge(image_template,image_target,steps):
     success,encode_image_temp = cv2.imencode(".jpg",image_template) #注意这句编码，否则图像质量不合格
     image_temp = base64.b64encode(encode_image_temp)
     success,encode_image_targ = cv2.imencode(".jpg",image_target) #注意这句编码，否则图像质量不合格
     image_targ = base64.b64encode(encode_image_targ)
     request_url = 'https://aip.baidubce.com/rest/2.0/face/v1/merge'
-    params = {
-        "image_template": {               # 将其中一张图片设置为模板（相当于底层图片）
-            "image": str(image_temp,'utf-8'),
-            "image_type": "BASE64",
-            "quality_control": "NORMAL"
-        },
-        "image_target": {                 # 将其中一张图片设置为目标（相当于人脸信息“叠加”到模板上）
-            "image": str(image_targ,'utf-8'),
-            "image_type": "BASE64",
-            "quality_control": "NORMAL"
-        },
-        "merge_degree": "HIGH",            # 融合程度
-        "position": 2,                     #左上角水印
-        "language": 1,                     #英语水印
-    }  
-    params=json.dumps(params)
     access_token = get_token('gZBF8TaiLfewLu4XdWIVe5H1', 'O7zkY3cc1OYu2GHwqvsyxI4YaCgdRn7A')
     request_url = request_url + "?access_token=" + access_token
     headers ={'content-type':'application/json'}
-    result = requests.post(request_url, data=params, headers=headers).json()  # 经过调用百度云接口服务器返回的内容（融合结果）
-    if result['error_code'] == 0:
-        res = result["result"]["merge_image"]
-        res = base64.b64decode(res)
-        nparr = np.frombuffer(res,np.uint8)
-        img_np = cv2.imdecode(nparr,cv2.IMREAD_COLOR)
-        return img_np
-    else:
-        print(str(result['error_code']) + result['error_msg'])
-        return result['error_code']
+    
+    for i in range(steps,-1,-1):
+        params = {
+            "image_template": {               # 将其中一张图片设置为模板（相当于底层图片）
+                "image": str(image_temp,'utf-8'),
+                "image_type": "BASE64",
+                "quality_control": "NORMAL"
+            },
+            "image_target": {                 # 将其中一张图片设置为目标（相当于人脸信息“叠加”到模板上）
+                "image": str(image_targ,'utf-8'),
+                "image_type": "BASE64",
+                "quality_control": "NORMAL"
+            },
+            "merge_degree": "HIGH",            # 融合程度
+            "position": 2,                     #左上角水印
+            "language": 1,                     #英语水印
+            "version": 4.0,
+            "alpha": i/steps,
+        }  
+        params=json.dumps(params)
+        result = requests.post(request_url, data=params, headers=headers).json()  # 经过调用百度云接口服务器返回的内容（融合结果）
+        if result['error_code'] == 0:
+            res = result["result"]["merge_image"]
+            res = base64.b64decode(res)
+            nparr = np.frombuffer(res,np.uint8)
+            img_np = cv2.imdecode(nparr,cv2.IMREAD_COLOR)
+            yield img_np
+        else:
+            print(str(result['error_code']) + result['error_msg'])
+            return result['error_code']
 
 
 with gr.Blocks() as demo:
     gr.Markdown("Face merge.")
 
     with gr.Row():
-        image_template = gr.Image()
-        image_target = gr.Image()
-        image_output = gr.Image()
+        image_template = gr.Image(label="模板图片")
+        image_target = gr.Image(label="目标图片")
+    with gr.Row():
+        steps = gr.Slider(1, 10, value=1, step=1, label="变化步数")
+        image_output = gr.Image(label="融合结果")
     image_button = gr.Button("Merge")
 
-    image_button.click(face_merge, inputs=[image_template,image_target], outputs=image_output)
+    image_button.click(face_merge, inputs=[image_template,image_target,steps], outputs=image_output)
 
 gr.close_all()
+demo.queue()
 demo.launch()