Abhishek Gola commited on
Commit
38043fd
·
1 Parent(s): 7eae67c

Added samples

Browse files
.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
@@ -54,14 +54,39 @@ def segment_person(input_image):
54
  return output
55
 
56
  # Gradio Interface
57
- demo = gr.Interface(
58
- fn=segment_person,
59
- inputs=gr.Image(type="numpy", label="Upload Image"),
60
- outputs=gr.Image(type="numpy", label="Human Segmentation Output"),
61
- title="Human Segmentation PPHumanSeg (OpenCV DNN)",
62
- allow_flagging="never",
63
- description="Upload an image to segment human regions using OpenCV's ONNX-based PPHumanSeg model."
64
- )
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
65
 
66
  if __name__ == "__main__":
67
  demo.launch()
 
54
  return output
55
 
56
  # Gradio Interface
57
+ with gr.Blocks(css='''.example * {
58
+ font-style: italic;
59
+ font-size: 18px !important;
60
+ color: #0ea5e9 !important;
61
+ }''') as demo:
62
+
63
+ gr.Markdown("### Human Segmentation PPHumanSeg (OpenCV DNN)")
64
+ gr.Markdown("Upload an image to segment human regions using OpenCV's ONNX-based PPHumanSeg model.")
65
+
66
+ with gr.Row():
67
+ input_image = gr.Image(type="numpy", label="Upload Image")
68
+ output_image = gr.Image(type="numpy", label="Human Segmentation Output")
69
+
70
+ # Clear output when new image is uploaded
71
+ input_image.change(fn=lambda: (None), outputs=output_image)
72
+
73
+ with gr.Row():
74
+ submit_btn = gr.Button("Submit", variant="primary")
75
+ clear_btn = gr.Button("Clear")
76
+
77
+ submit_btn.click(fn=segment_person, inputs=input_image, outputs=output_image)
78
+ clear_btn.click(fn=lambda:(None, None), outputs=[input_image, output_image])
79
+
80
+ gr.Markdown("Click on any example to try it.", elem_classes=["example"])
81
+
82
+ gr.Examples(
83
+ examples=[
84
+ ["examples/googlenet_0.png"],
85
+ ["examples/gray_face.png"],
86
+ ["examples/messi5.jpg"]
87
+ ],
88
+ inputs=input_image
89
+ )
90
 
91
  if __name__ == "__main__":
92
  demo.launch()
examples/googlenet_0.png ADDED

Git LFS Details

  • SHA256: 4a29f6a95dea309d5cf4e429e732df964107cea11f620a5590289354cdecb456
  • Pointer size: 130 Bytes
  • Size of remote file: 90.2 kB
examples/gray_face.png ADDED

Git LFS Details

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

Git LFS Details

  • SHA256: 1d570e49654e84c7a943918537bd9e5e1ef82920152e147c834006e235be97c9
  • Pointer size: 130 Bytes
  • Size of remote file: 72.9 kB