gradio/demo/stop_sign_detection.py

23 lines
489 B
Python
Raw Normal View History

2020-11-24 07:11:37 +08:00
# Demo: (Image) -> (Image)
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()