init

app.py

@@ -1,14 +1,32 @@
 import cv2
 import gradio as gr
+import numpy as np
+
 from z_app_factory import get_app
 
+thickness = 3
+lineType = 8
+font = cv2.FONT_HERSHEY_SIMPLEX
 
 def inference(image):
     image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
-    lst2d_res = get_app(image)
-    thickness = 3
-    lineType = 8
-    font = cv2.FONT_HERSHEY_SIMPLEX
+    code,lst2d_res = get_app(image)
+    if code == 401:
+        msg = "Not RGB three channel picture"
+    elif code == 402:
+        msg = "Pixels less than 32 × 32"
+    elif code == 403:
+        msg = "Pixels greater than 4096 × 4096"
+    elif code == 404:
+        msg = "Files larger than 5MB"
+    elif code == 405:
+        msg = "System error, please contact the server for troubleshooting"
+
+    if code!=200:
+        img_out = np.zeros((500, 600,3))
+        cv2.putText(img_out, msg, (20, 20), font, 1, (0, 255, 0), 2)
+        return img_out
+
 
     for face in lst2d_res:
         bbox = [int(i) for i in face["bbox"]]
@@ -26,20 +44,14 @@ def inference(image):
             x, y = [int(i) for i in kp]
             cv2.circle(image, (x, y), 2, (2, 30, 200), 2)
 
-        landmarks = face["landmarks"]
-        lst = [[int(i) for i in kp] for kp in landmarks]
-        for i, kp in enumerate(lst):
-            x, y = kp
-            cv2.circle(image, (x, y), 1, (200, 200, 20), 1)
-
     image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
     return image
 
 
 
-title = "Face Keypoint"
-description = "demo for Face Keypoint. To use it, simply upload your image, or click one of the examples to load them. Read more at the links below."
-article = "<p style='text-align: center'><a href='https://www.yuque.com/itmorn/ability/face_keypoint' target='_blank'>Project Documents</a> | <a href='https://www.bilibili.com/video/BV1DN4y1P7j3' target='_blank'>Video Demo</a></p>"
+title = "Detect Face"
+description = "demo for Detect Face. To use it, simply upload your image, or click one of the examples to load them. Read more at the links below."
+article = "<p style='text-align: center'><a href='https://www.yuque.com/itmorn/ability/detect_face' target='_blank'>Project Documents</a> | <a href='https://www.bilibili.com/video/BV1cg411v7CR' target='_blank'>Video Demo</a></p>"
 
 gr.Interface(
     inference,
@@ -51,5 +63,5 @@ gr.Interface(
     examples=[
               ["imgs/face1.jpg"],
               ["imgs/face2.jpg"],
-              ["imgs/cc.png"]
+              ["imgs/11.jpg"]
     ]).launch(debug=True)

core.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:28a28f235c49290d48d6acb21ddacbb3cd44a3e84cb37d5bb8dacf7d8eeb8120
-size 20954255
+oid sha256:e76e20bf4de2ba09de06c1af157ac4495e94d24d068622fc64b9fad49514567d
+size 20956224