File size: 1,833 Bytes
c6f75b2 9dd9d3c c6f75b2 99e17b1 59f3cda 99e17b1 59f3cda 99e17b1 c6f75b2 cc86420 1e0c1a0 cc86420 99e17b1 1e0c1a0 c6f75b2 f8b6d7c c6f75b2 f8b6d7c c6f75b2 3da5743 c6f75b2 |
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 |
import gradio as gr
import cv2
import requests
import os
from ultralytics import YOLO
model = YOLO('best.pt')
def normalize_preds(label):
if label =="Blank":
label = "Deer"
return label
def show_preds_image(image):
# Save the uploaded image temporarily
image_path = "uploaded_image.jpg"
cv2.imwrite(image_path, image[:, :, ::-1]) # Convert BGR to RGB and save the image
image = cv2.imread(image_path)
outputs = model.predict(source=image_path)
# print("output:: ", type(outputs[0]))
names = outputs[0].names
results = outputs[0].boxes.xyxy.cpu().numpy()
probs = outputs[0].boxes.conf.cpu().numpy()
species = outputs[0].boxes.cls.cpu().numpy()
# print(probs, names)
for idx in range(len(results)):
x1, y1, x2, y2 = results[idx][:4]
# print(type(det),det)
label = normalize_preds(names[int(species[idx])])
prob_cls = probs[idx]
cv2.rectangle(
image,
(int(x1), int(y1)),
(int(x2), int(y2)),
color=(0, 0, 255),
thickness=4,
lineType=cv2.LINE_AA,
)
cv2.putText(
image,
label,
(int(x1), int(y1) + 20),
cv2.FONT_HERSHEY_SIMPLEX,
0.9,
(0, 0, 255),
2,
cv2.LINE_AA,
)
os.remove(image_path) # Remove the temporary image file
return cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
inputs_image = [
gr.inputs.Image(label="Upload Image"),
]
outputs_image = [
gr.outputs.Image(type="numpy"),
]
interface_image = gr.Interface(
fn=show_preds_image,
inputs=inputs_image,
outputs=outputs_image,
title="ReWilding Europe Project - Demo[Yolov8]",
examples=[],
cache_examples=False,
)
interface_image.launch()
|