Update

app.py

@@ -5,13 +5,14 @@ import json
 from PIL import Image
 
 def get_attributes(json):    
+    print("1------------------")
     liveness = "GENUINE" if json.get('liveness') >= 0.5 else "FAKE"
     attr = json.get('attribute')
     age = attr.get('age')
     gender = attr.get('gender')
     emotion = attr.get('emotion')
     ethnicity = attr.get('ethnicity')
-
+    print("2------------------")
     mask = [attr.get('face_mask')]
     if attr.get('glasses') == 'USUAL':
         mask.append('GLASSES')
@@ -23,13 +24,15 @@ def get_attributes(json):
         eye.append('LEFT')
     if attr.get('eye_right') >= 0.5:
         eye.append('RIGHT')
-
+    print("3------------------")
     facehair = attr.get('facial_hair')
     haircolor = attr.get('hair_color')
     hairtype = attr.get('hair_type')
     headwear = attr.get('headwear')
-    
+    print("4------------------")
     activity = []
+    if attr.get('food_consumption') >= 0.5:
+        activaty.append('EATING')
     if attr.get('phone_recording') >= 0.5:
         activaty.append('PHONE_RECORDING')
     if attr.get('phone_use') >= 0.5:
@@ -38,12 +41,12 @@ def get_attributes(json):
         activaty.append('SEATBELT')
     if attr.get('smoking') >= 0.5:
         activaty.append('SMOKING')
-
+    print("5------------------")
     pitch = attr.get('pitch')
     roll = attr.get('roll')
     yaw = attr.get('yaw')
     quality = attr.get('quality')
-
+    print("6------------------")
     return liveness, age, gender, emotion, ethnicity, mask, eye, facehair, haircolor, hairtype, headwear, activity, pitch, roll, yaw, quality
 
 def compare_face(frame1, frame2):
@@ -60,6 +63,7 @@ def compare_face(frame1, frame2):
 
         face1 = Image.new('RGBA',(150, 150), (80,80,80,0))
         face2 = Image.new('RGBA',(150, 150), (80,80,80,0))
+
         liveness1, age1, gender1, emotion1, ethnicity1, mask1, eye1, facehair1, haircolor1, hairtype1, headwear1, activity1, pitch1, roll1, yaw1, quality1 = [None] * 16
         liveness2, age2, gender2, emotion2, ethnicity2, mask2, eye2, facehair2, haircolor2, hairtype2, headwear2, activity2, pitch2, roll2, yaw2, quality2 = [None] * 16
         res1 = r.json().get('image1')