Turing311 commited on
Commit
4bee2af
·
1 Parent(s): 74d3a4b
Files changed (1) hide show
  1. app.py +19 -20
app.py CHANGED
@@ -62,9 +62,7 @@ def compare_face(frame1, frame2):
62
  resized_h = 150
63
 
64
  face1 = face1.resize((int(resized_w), int(resized_h)))
65
- print("################", face1)
66
  liveness1, age1, gender1, emotion1, ethnicity1, mask1, eye1, facehair1, haircolor1, hairtype1, headwear1, activity1, pitch1, roll1, yaw1, quality1 = get_attributes(res1)
67
- print("@@@@@@@@@@@@@@@@", liveness1, age1)
68
 
69
  res2 = r.json().get('image2')
70
  if res2 is not None:
@@ -132,8 +130,9 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue=primary_color, secondary_hue=
132
  with gr.Column(variant="panel"):
133
  gr.Markdown("<b>Image 1<b/>")
134
  liveness1 = gr.CheckboxGroup(["GENUINE", "FAKE"], label="Liveness")
135
- age1 = gr.Number(0, label="Age")
136
- gender1 = gr.CheckboxGroup(["MALE", "FEMALE"], label="Gender")
 
137
  emotion1 = gr.CheckboxGroup(["HAPPINESS", "ANGER", "FEAR", "NEUTRAL", "SADNESS", "SURPRISE"], label="Emotion")
138
  ethnicity1 = gr.CheckboxGroup(["ASIAN", "BLACK", "CAUCASIAN", "EAST_INDIAN"], label="Ethnicity")
139
  mask1 = gr.CheckboxGroup(["LOWER_FACE_MASK", "FULL_FACE_MASK", "OTHER_MASK", "GLASS", "SUNGLASS"], label="Mask & Glasses")
@@ -150,23 +149,23 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue=primary_color, secondary_hue=
150
  quality1 = gr.Number(0, label="Quality")
151
  with gr.Column(variant="panel"):
152
  gr.Markdown("<b>Image 2<b/>")
153
- liveness2 = gr.CheckboxGroup(["GENUINE", "FAKE"], value="GENUINE", label="Liveness")
154
- age2 = gr.Number(2, label="Age")
155
- gender2 = gr.CheckboxGroup(["MALE", "FEMALE"], value="MALE", label="Gender")
156
- emotion2 = gr.CheckboxGroup(["HAPPINESS", "ANGER", "FEAR", "NEUTRAL", "SADNESS", "SURPRISE"], value="HAPPINESS", label="Emotion")
157
- ethnicity2 = gr.CheckboxGroup(["ASIAN", "BLACK", "CAUCASIAN", "EAST_INDIAN"], value=["EAST_INDIAN"], label="Ethnicity")
158
- mask2 = gr.CheckboxGroup(["LOWER_FACE_MASK", "FULL_FACE_MASK", "OTHER_MASK", "GLASS", "SUNGLASS"], value=["MASK"], label="Mask & Glasses")
159
- eye2 = gr.CheckboxGroup(["LEFT", "RIGHT"], value=["LEFT"], label="Eye Open")
160
- facehair2 = gr.CheckboxGroup(["BEARD", "BRISTLE", "MUSTACHE", "SHAVED"], value=["BEARD"], label="Facial Hair")
161
- haircolor2 = gr.CheckboxGroup(["BLACK", "BLOND", "BROWN"], value=["BLACK"], label="Hair Color")
162
- hairtype2 = gr.CheckboxGroup(["BALD", "SHORT", "MEDIUM", "LONG"], value=["MEDIUM"], label="Hair Type")
163
- headwear2 = gr.CheckboxGroup(["B_CAP", "CAP", "HAT", "HELMET", "HOOD"], value=["NO_MASK"], label="Head Wear")
164
- activity2 = gr.CheckboxGroup(["EATING", "PHONE_RECORDING", "PHONE_USE", "SMOKING", "SEATBELT"], value=["NO_MASK"], label="Activity")
165
  with gr.Row():
166
- pitch2 = gr.Number(2.14, label="Pitch")
167
- roll2 = gr.Number(1.33, label="Roll")
168
- yaw2 = gr.Number(0.63, label="Yaw")
169
- quality2 = gr.Number(0.63, label="Quality")
170
 
171
  compare_result_output = gr.JSON(label='Result')
172
 
 
62
  resized_h = 150
63
 
64
  face1 = face1.resize((int(resized_w), int(resized_h)))
 
65
  liveness1, age1, gender1, emotion1, ethnicity1, mask1, eye1, facehair1, haircolor1, hairtype1, headwear1, activity1, pitch1, roll1, yaw1, quality1 = get_attributes(res1)
 
66
 
67
  res2 = r.json().get('image2')
68
  if res2 is not None:
 
130
  with gr.Column(variant="panel"):
131
  gr.Markdown("<b>Image 1<b/>")
132
  liveness1 = gr.CheckboxGroup(["GENUINE", "FAKE"], label="Liveness")
133
+ with gr.Row():
134
+ age1 = gr.Number(0, label="Age")
135
+ gender1 = gr.CheckboxGroup(["MALE", "FEMALE"], label="Gender")
136
  emotion1 = gr.CheckboxGroup(["HAPPINESS", "ANGER", "FEAR", "NEUTRAL", "SADNESS", "SURPRISE"], label="Emotion")
137
  ethnicity1 = gr.CheckboxGroup(["ASIAN", "BLACK", "CAUCASIAN", "EAST_INDIAN"], label="Ethnicity")
138
  mask1 = gr.CheckboxGroup(["LOWER_FACE_MASK", "FULL_FACE_MASK", "OTHER_MASK", "GLASS", "SUNGLASS"], label="Mask & Glasses")
 
149
  quality1 = gr.Number(0, label="Quality")
150
  with gr.Column(variant="panel"):
151
  gr.Markdown("<b>Image 2<b/>")
152
+ liveness2 = gr.CheckboxGroup(["GENUINE", "FAKE"], label="Liveness")
153
+ age2 = gr.Number(0, label="Age")
154
+ gender2 = gr.CheckboxGroup(["MALE", "FEMALE"], label="Gender")
155
+ emotion2 = gr.CheckboxGroup(["HAPPINESS", "ANGER", "FEAR", "NEUTRAL", "SADNESS", "SURPRISE"], label="Emotion")
156
+ ethnicity2 = gr.CheckboxGroup(["ASIAN", "BLACK", "CAUCASIAN", "EAST_INDIAN"], label="Ethnicity")
157
+ mask2 = gr.CheckboxGroup(["LOWER_FACE_MASK", "FULL_FACE_MASK", "OTHER_MASK", "GLASS", "SUNGLASS"], label="Mask & Glasses")
158
+ eye2 = gr.CheckboxGroup(["LEFT", "RIGHT"], label="Eye Open")
159
+ facehair2 = gr.CheckboxGroup(["BEARD", "BRISTLE", "MUSTACHE", "SHAVED"], label="Facial Hair")
160
+ haircolor2 = gr.CheckboxGroup(["BLACK", "BLOND", "BROWN"], label="Hair Color")
161
+ hairtype2 = gr.CheckboxGroup(["BALD", "SHORT", "MEDIUM", "LONG"], label="Hair Type")
162
+ headwear2 = gr.CheckboxGroup(["B_CAP", "CAP", "HAT", "HELMET", "HOOD"], label="Head Wear")
163
+ activity2 = gr.CheckboxGroup(["EATING", "PHONE_RECORDING", "PHONE_USE", "SMOKING", "SEATBELT"], label="Activity")
164
  with gr.Row():
165
+ pitch2 = gr.Number(0, label="Pitch")
166
+ roll2 = gr.Number(0, label="Roll")
167
+ yaw2 = gr.Number(0, label="Yaw")
168
+ quality2 = gr.Number(0, label="Quality")
169
 
170
  compare_result_output = gr.JSON(label='Result')
171