| import gradio as gr |
| from gradio_image_annotation import image_annotator |
| import time |
| import numpy as np |
| import random |
|
|
| example_annotation = { |
| "image": "https://gradio-builds.s3.amazonaws.com/demo-files/base.png", |
| "boxes": [ |
| { |
| "xmin": 636, |
| "ymin": 575, |
| "xmax": 801, |
| "ymax": 697, |
| "label": "Vehicle", |
| "color": (255, 0, 0) |
| }, |
| { |
| "xmin": 360, |
| "ymin": 615, |
| "xmax": 386, |
| "ymax": 702, |
| "label": "Person", |
| "color": (0, 255, 0) |
| } |
| ] |
| } |
|
|
| example_crop = { |
| "image": "https://raw.githubusercontent.com/gradio-app/gradio/main/guides/assets/logo.png", |
| "boxes": [ |
| { |
| "xmin": 30, |
| "ymin": 70, |
| "xmax": 530, |
| "ymax": 500, |
| "color": (100, 200, 255) |
| } |
| ] |
| } |
|
|
|
|
| def run(annotations): |
| while True: |
| w = 1920 |
| |
| h = 1080 |
| img = np.random.random((h, w, 3)) |
| yield { |
| "image": img |
| } |
| time.sleep(1) |
|
|
| def run2(annotations): |
| while True: |
| w = 1920 |
| |
| h = 1080 |
| img = np.random.random((h, w, 3)) |
| yield img |
| time.sleep(1) |
|
|
| with gr.Blocks() as demo: |
| with gr.Row(): |
| annotator = image_annotator( |
| example_crop, |
| image_type="numpy", |
| ) |
| image = gr.Image() |
| button = gr.Button() |
| button.click(run, annotator, annotator) |
| button2 = gr.Button() |
| button2.click(run2, image, image) |
|
|
|
|
| if __name__ == "__main__": |
| demo.launch() |
|
|
