| import gradio as gr |
| from utils import object_detection_brainai as odb |
| object_detection = odb.ObjectDetectionModel() |
|
|
| def detect_image(img): |
| result_img = object_detection.process(img) |
|
|
| return result_img |
|
|
| def detect_video(video_path): |
| output_video_path = object_detection.play_video(video_path) |
|
|
| return output_video_path |
|
|
| image_interface = gr.Interface( |
| fn=detect_image, |
| title="Image Object Detection", |
| inputs=gr.Image(label="Upload Image"), |
| outputs="image", |
| examples = ['data/office.jpg', 'data/classroom.jpg']) |
|
|
| video_interface = gr.Interface( |
| fn=detect_video, |
| inputs=gr.Video(label="Upload Video"), |
| examples = ['data/breakfast-example.mp4'], |
| outputs="video", |
| title="Video Object Detection") |
|
|
| demo = gr.TabbedInterface([image_interface, video_interface], ["Image", "Video"]) |
|
|
| demo.launch() |
|
|
