|
import gradio as gr |
|
from gradio_image_annotation import image_annotator |
|
|
|
example = { |
|
"image": "https://raw.githubusercontent.com/gradio-app/gradio/main/guides/assets/logo.png", |
|
"boxes": [ |
|
{ |
|
"xmin": 30, |
|
"ymin": 70, |
|
"xmax": 530, |
|
"ymax": 500, |
|
"label": "Gradio", |
|
"color": (250, 185, 0), |
|
} |
|
] |
|
} |
|
|
|
|
|
def crop(annotations): |
|
if annotations["boxes"]: |
|
box = annotations["boxes"][0] |
|
return annotations["image"][ |
|
box["ymin"]:box["ymax"], |
|
box["xmin"]:box["xmax"] |
|
] |
|
return None |
|
|
|
|
|
def get_boxes_json(annotations): |
|
return annotations["boxes"] |
|
|
|
with gr.Blocks() as demo: |
|
with gr.Tab("Object annotation"): |
|
annotator = image_annotator( |
|
{"image": "https://gradio-builds.s3.amazonaws.com/demo-files/base.png"}, |
|
label_list=["Person", "Vehicle"], |
|
label_colors=[(0, 255, 0), (255, 0, 0)], |
|
|
|
handle_size=6, |
|
box_thickness=0, |
|
box_selected_thickness=1, |
|
) |
|
button_get = gr.Button("Get bounding boxes") |
|
json_boxes = gr.JSON() |
|
button_get.click(get_boxes_json, annotator, json_boxes) |
|
with gr.Tab("Crop"): |
|
with gr.Row(): |
|
annotator_crop = image_annotator(example, image_type="numpy", |
|
handle_size=6, |
|
box_thickness=1, |
|
box_selected_thickness=1,) |
|
image_crop = gr.Image() |
|
button_crop = gr.Button("Crop") |
|
button_crop.click(crop, annotator_crop, image_crop) |
|
|
|
|
|
if __name__ == "__main__": |
|
demo.launch() |
|
|