|
import gradio as gr |
|
import numpy as np |
|
|
|
from ultralyticsplus import YOLO, render_result |
|
import cv2 |
|
from PIL import Image |
|
|
|
from cv2 import imshow |
|
from cv2 import imwrite |
|
|
|
|
|
|
|
|
|
|
|
def PPE(image): |
|
|
|
|
|
model = YOLO('keremberke/yolov8m-hard-hat-detection') |
|
|
|
model.overrides['conf'] = 0.25 |
|
model.overrides['iou'] = 0.45 |
|
model.overrides['agnostic_nms'] = False |
|
model.overrides['max_det'] = 1000 |
|
|
|
|
|
|
|
|
|
results = model.predict(image) |
|
|
|
|
|
print(results[0].boxes) |
|
render = render_result(model=model, image=image, result=results[0]) |
|
render.show() |
|
|
|
|
|
|
|
|
|
|
|
|
|
demo = gr.Interface( |
|
PPE, |
|
gr.Image(source="webcam", streaming=True), |
|
"image", |
|
live=True |
|
) |
|
demo.launch() |