Abhishek Gola commited on
Commit
f042d39
·
1 Parent(s): c6bbca4

Added samples

Browse files
Files changed (4) hide show
  1. .gitattributes +5 -0
  2. app.py +39 -9
  3. examples/left.jpg +3 -0
  4. examples/messi5.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
@@ -77,15 +77,45 @@ def detect_objects(input_image):
77
  vis_image = visualize(results, bgr, scale)
78
  return cv.cvtColor(vis_image, cv.COLOR_BGR2RGB)
79
 
80
- # Gradio interface
81
- demo = gr.Interface(
82
- fn=detect_objects,
83
- inputs=gr.Image(type="numpy", label="Upload Image"),
84
- outputs=gr.Image(type="numpy", label="Detected Objects"),
85
- title="YOLOX Object Detection (OpenCV + ONNX)",
86
- description="Upload an image to detect objects using YOLOX ONNX model and OpenCV DNN.",
87
- allow_flagging="never"
88
- )
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
89
 
90
  if __name__ == "__main__":
91
  demo.launch()
 
77
  vis_image = visualize(results, bgr, scale)
78
  return cv.cvtColor(vis_image, cv.COLOR_BGR2RGB)
79
 
80
+ def clear_all():
81
+ return None, None
82
+
83
+ def clear_output():
84
+ return None
85
+
86
+ with gr.Blocks(css='''.example * {
87
+ font-style: italic;
88
+ font-size: 18px !important;
89
+ color: #0ea5e9 !important;
90
+ }''') as demo:
91
+
92
+ gr.Markdown("### YOLOX Object Detection (OpenCV + ONNX)")
93
+ gr.Markdown("Upload an image to detect objects using YOLOX ONNX model and OpenCV DNN.")
94
+
95
+ with gr.Row():
96
+ image_input = gr.Image(type="numpy", label="Upload Image")
97
+ output_image = gr.Image(type="numpy", label="Detected Objects")
98
+
99
+ # Clear output when new image is uploaded
100
+ image_input.change(fn=clear_output, outputs=[output_image])
101
+
102
+ with gr.Row():
103
+ submit_btn = gr.Button("Submit", variant="primary")
104
+ clear_btn = gr.Button("Clear")
105
+
106
+ submit_btn.click(fn=detect_objects, inputs=image_input, outputs=output_image)
107
+ clear_btn.click(fn=clear_all, outputs=[image_input, output_image])
108
+
109
+ gr.Markdown("Click on any example to try it.", elem_classes=["example"])
110
+
111
+ gr.Examples(
112
+ examples=[
113
+ ["examples/left.jpg"],
114
+ ["examples/messi5.jpg"]
115
+ ],
116
+ inputs=image_input
117
+ )
118
+
119
 
120
  if __name__ == "__main__":
121
  demo.launch()
examples/left.jpg ADDED

Git LFS Details

  • SHA256: fb314330c3eb0a81651c682803e73b348898157267fbc37c4c2c07d1b3a8e321
  • Pointer size: 130 Bytes
  • Size of remote file: 24.1 kB
examples/messi5.jpg ADDED

Git LFS Details

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