Abhishek Gola commited on
Commit
5699031
·
1 Parent(s): c741d15

Added samples

Browse files
Files changed (4) hide show
  1. .gitattributes +5 -0
  2. app.py +34 -11
  3. examples/gray_face.png +3 -0
  4. examples/lena.jpg +3 -0
.gitattributes CHANGED
@@ -14,6 +14,11 @@
14
  *.npy filter=lfs diff=lfs merge=lfs -text
15
  *.npz filter=lfs diff=lfs merge=lfs -text
16
  *.onnx filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
17
  *.ot filter=lfs diff=lfs merge=lfs -text
18
  *.parquet filter=lfs diff=lfs merge=lfs -text
19
  *.pb filter=lfs diff=lfs merge=lfs -text
 
14
  *.npy filter=lfs diff=lfs merge=lfs -text
15
  *.npz filter=lfs diff=lfs merge=lfs -text
16
  *.onnx filter=lfs diff=lfs merge=lfs -text
17
+ *.mp4 filter=lfs diff=lfs merge=lfs -text
18
+ *.jpg filter=lfs diff=lfs merge=lfs -text
19
+ *.gif filter=lfs diff=lfs merge=lfs -text
20
+ *.png filter=lfs diff=lfs merge=lfs -text
21
+ *.webp filter=lfs diff=lfs merge=lfs -te
22
  *.ot filter=lfs diff=lfs merge=lfs -text
23
  *.parquet filter=lfs diff=lfs merge=lfs -text
24
  *.pb filter=lfs diff=lfs merge=lfs -text
app.py CHANGED
@@ -76,17 +76,40 @@ def assess_face_quality(input_image):
76
  return f"Quality Score: {score:.3f}", output_image
77
 
78
  # Gradio Interface
79
- demo = gr.Interface(
80
- fn=assess_face_quality,
81
- inputs=gr.Image(type="numpy", label="Upload Face Image"),
82
- outputs=[
83
- gr.Text(label="Quality Score"),
84
- gr.Image(type="numpy", label="Aligned Face with Score")
85
- ],
86
- title="Face Image Quality Assessment (eDifFIQA + YuNet)",
87
- allow_flagging="never",
88
- description="Upload a face image. The app detects and aligns the face, then evaluates image quality using the eDifFIQA model."
89
- )
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
90
 
91
  if __name__ == "__main__":
92
  demo.launch()
 
76
  return f"Quality Score: {score:.3f}", output_image
77
 
78
  # Gradio Interface
79
+ with gr.Blocks(css='''.example * {
80
+ font-style: italic;
81
+ font-size: 18px !important;
82
+ color: #0ea5e9 !important;
83
+ }''') as demo:
84
+
85
+ gr.Markdown("### Face Image Quality Assessment (eDifFIQA + YuNet)")
86
+ gr.Markdown("Upload a face image. The app detects and aligns the face, then evaluates image quality using the eDifFIQA model.")
87
+
88
+ with gr.Row():
89
+ input_image = gr.Image(type="numpy", label="Upload Face Image")
90
+ with gr.Column():
91
+ quality_score = gr.Text(label="Quality Score")
92
+ aligned_face = gr.Image(type="numpy", label="Aligned Face with Score")
93
+
94
+ # Clear output when new image is uploaded
95
+ input_image.change(fn=lambda: ("", None), outputs=[quality_score, aligned_face])
96
+
97
+ with gr.Row():
98
+ submit_btn = gr.Button("Submit", variant="primary")
99
+ clear_btn = gr.Button("Clear")
100
+
101
+ submit_btn.click(fn=assess_face_quality, inputs=input_image, outputs=[quality_score, aligned_face])
102
+ clear_btn.click(fn=lambda: (None, "", None), outputs=[input_image, quality_score, aligned_face])
103
+
104
+ gr.Markdown("Click on any example to try it.", elem_classes=["example"])
105
+
106
+ gr.Examples(
107
+ examples=[
108
+ ["examples/lena.jpg"],
109
+ ["examples/gray_face.png"]
110
+ ],
111
+ inputs=input_image
112
+ )
113
 
114
  if __name__ == "__main__":
115
  demo.launch()
examples/gray_face.png ADDED

Git LFS Details

  • SHA256: 6df8d32e7771e2a5fa9a7f00d8d32d1e69e969ce907b5825920bbdb6962d8a45
  • Pointer size: 130 Bytes
  • Size of remote file: 40.1 kB
examples/lena.jpg ADDED

Git LFS Details

  • SHA256: 7de7ed51a1594fff247f4cae2301eceacf5313d6011e37b4a4c8733f7bb72c07
  • Pointer size: 130 Bytes
  • Size of remote file: 91.8 kB