| | 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() |
| |
|
