File size: 1,998 Bytes
f22bcea 857e0c5 f22bcea ac029d1 f22bcea 857e0c5 f22bcea 857e0c5 f22bcea 857e0c5 f22bcea 857e0c5 f22bcea a79dfef f22bcea |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 |
import supervision as sv
import gradio as gr
from ultralytics import YOLO
import sahi
import numpy as np
# Images
sahi.utils.file.download_from_url(
"https://www.erbanotizie.com/wp-content/uploads/2014/01/Casello.jpg",
"ocr1.jpg",
)
sahi.utils.file.download_from_url(
"https://media-cdn.tripadvisor.com/media/photo-s/15/1d/03/18/receipt.jpg",
"ocr2.jpg",
)
sahi.utils.file.download_from_url(
"https://upload.forumfree.net/i/ff11450850/b5ef33b7-01da-4055-9ece-089b2a35a193.jpg",
"ocr3.jpg",
)
annotatorbbox = sv.BoxAnnotator()
annotatormask=sv.MaskAnnotator()
model = YOLO("best_Receipt.pt")
def yolov8_inference(
image: gr.inputs.Image = None,
conf_threshold: gr.inputs.Slider = 0.5,
iou_threshold: gr.inputs.Slider = 0.45,
):
image=image[:, :, ::-1].astype(np.uint8)
model = YOLO("https://huggingface.co/spaces/devisionx/first-demo/blob/main/best_Receipt.pt")
results = model(image,imgsz=320)[0]
image=image[:, :, ::-1].astype(np.uint8)
detections = sv.Detections.from_yolov8(results)
annotated_image = annotatormask.annotate(scene=image, detections=detections)
annotated_image = annotatorbbox.annotate(scene=annotated_image , detections=detections)
return annotated_image
image_input = gr.inputs.Image() # Adjust the shape according to your requirements
inputs = [
gr.inputs.Image(label="Input Image"),
gr.Slider(
minimum=0.0, maximum=1.0, value=0.25, step=0.05, label="Confidence Threshold"
),
gr.Slider(minimum=0.0, maximum=1.0, value=0.45, step=0.05, label="IOU Threshold"),
]
outputs = gr.Image(type="filepath", label="Output Image")
title = "OCR Demo"
examples = [
["ocr1.jpg", 0.6, 0.45],
["ocr2.jpg", 0.25, 0.45],
["ocr3.jpg", 0.25, 0.45],
]
demo_app = gr.Interface(examples=examples,
fn=yolov8_inference,
inputs=inputs,
outputs=outputs,
title=title,
cache_examples=True,
theme="default",
)
demo_app.launch(debug=False, enable_queue=True) |