edgargg's picture
Upload folder using huggingface_hub
29580f3 verified
raw
history blame
1.56 kB
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 [
{k: box[k]
for k in box if k in ("xmin", "ymin", "xmax", "ymax", "label")}
for box in annotations["boxes"]
]
with gr.Blocks() as demo:
with gr.Tab("Crop"):
with gr.Row():
annotator_crop = image_annotator(example, image_type="numpy")
image_crop = gr.Image()
button_crop = gr.Button("Crop")
button_crop.click(crop, annotator_crop, image_crop)
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)],
)
button_get = gr.Button("Get bounding boxes")
json_boxes = gr.JSON()
button_get.click(get_boxes_json, annotator, json_boxes)
if __name__ == "__main__":
demo.launch()