gradio/demo/stop_sign_detection.py

21 lines
461 B
Python
Raw Normal View History

2021-05-24 23:31:44 +08:00
import gradio as gr
def detect(image):
# return image
return (image, [("sign", 210, 80, 280, 150), ("train", 100, 100, 180, 150)])
iface = gr.Interface(detect,
gr.inputs.Image(type="pil"),
# "image",
"segmented_image",
examples=[
["images/stop_1.jpg"],
["images/stop_2.jpg"],
])
iface.test_launch()
if __name__ == "__main__":
iface.launch()