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