big face

.gitignore

@@ -0,0 +1 @@
+/.idea

app.py

@@ -3,32 +3,49 @@ import gradio as gr
 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)
-    annotated_image = image.copy()
-    lst2d_res = get_app(image)
-    thickness = 3
-    lineType = 8
-    font = cv2.FONT_HERSHEY_SIMPLEX
+    code,lst2d_res = get_app(image)
+    msg = "aaa"
+    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 greater than 5MB"
+    elif code == 405:
+        msg = "System error, please contact\n the server for troubleshooting"
+
+    if code!=200:
+        img_out = np.zeros((500, 600,3),dtype=np.uint8)
+        cv2.putText(img_out, msg, (20, 200), font, 1, (0, 255, 0), 2)
+        return img_out
 
     for face in lst2d_res:
+
         bbox = [int(i) for i in face["bbox"]]
         score = face['score']
         point_color = (0, int(255 * score), 0) # BGR
         x1, y1 = bbox[:2]
         x2, y2 = bbox[2:]
-        cv2.putText(annotated_image, str(score)[:4], (x1, y1 - 10), font, 0.8, (0, 255, 0), 2)
-        cv2.line(annotated_image, (x1, y1), (x2, y1), point_color, thickness, lineType)
-        cv2.line(annotated_image, (x2, y1), (x2, y2), point_color, thickness, lineType)
-        cv2.line(annotated_image, (x1, y1), (x1, y2), point_color, thickness, lineType)
-        cv2.line(annotated_image, (x1, y2), (x2, y2), point_color, thickness, lineType)
+        cv2.putText(image, str(score)[:4], (x1, y1 - 10), font, 0.8, (0, 255, 0), 2)
+        cv2.line(image, (x1, y1), (x2, y1), point_color, thickness, lineType)
+        cv2.line(image, (x2, y1), (x2, y2), point_color, thickness, lineType)
+        cv2.line(image, (x1, y1), (x1, y2), point_color, thickness, lineType)
+        cv2.line(image, (x1, y2), (x2, y2), point_color, thickness, lineType)
 
         for kp in face["kps"]:
             x, y = [int(i) for i in kp]
-            cv2.circle(annotated_image, (x, y), 2, (2, 30, 200), 2)
+            cv2.circle(image, (x, y), 2, (2, 30, 200), 2)
 
-    annotated_image = cv2.cvtColor(annotated_image, cv2.COLOR_RGB2BGR)
-    return annotated_image
+    image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
+    return image
 
 
 
@@ -46,5 +63,6 @@ gr.Interface(
     examples=[
               ["imgs/face1.jpg"],
               ["imgs/face2.jpg"],
-              ["imgs/11.jpg"]
+              ["imgs/cc.png"],
+              ["imgs/2313.png"]
     ]).launch(debug=True)

core.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1e8f1ea11748c0e7ee3b9e3f127348017f22fc576c4558e8ce4a6ec5bd315fe6
-size 16334604
+oid sha256:f02f1f3a3063a53802bd427234db4eed6dbc600e1e5dba59a9eb064401c40fc7
+size 20928518

mic.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ebc1ac4d0f2363419842427822b3ab81e3ba3d8eaf6da7a61800c3cbbc084991
-size 1367712
+oid sha256:2c5c351ce535def9746f385c8d7cf48e2d332ab4de71a0bbdbb40d0f82129524
+size 1926896