deneme

app.py

@@ -0,0 +1,55 @@
+import gradio as gr
+from ultralytics import YOLO
+from collections import Counter
+import torch
+import cv2
+
+# torch.hub.download_url_to_file('https://ultralytics.com/images/zidane.jpg', 'zidane.jpg')
+# torch.hub.download_url_to_file('https://ultralytics.com/images/bus.jpg', 'bus.jpg')
+
+
+def yolov8_predictor(img: gr.inputs.Image = None,
+                     model_path: gr.inputs.Dropdown = None,
+                     image_size= 640,
+                     conf= 0.25,
+                     iou= 0.7
+                    ):
+
+    model= YOLO(model= model_path)
+    res= model.predict(source= img, imgsz=image_size, conf= conf, iou= iou)
+
+    cls_names= res[0].names
+    dets_cls= res[0].boxes.cls.cpu().numpy().astype(int)
+
+    dets= [cls_names[i] for i in dets_cls]
+    dets= dict(Counter(dets))
+    res_plotted = res[0].plot()
+    res_plotted= cv2.cvtColor(res_plotted, cv2.COLOR_BGR2RGB)
+
+    return res_plotted, str(dets)
+
+inputs = [
+    gr.inputs.Image(type="filepath", label="Input Image"),
+    gr.inputs.Dropdown(["yolov8n.pt", "yolov8s.pt"], 
+                       default="yolov8s.pt", label="Model"),
+    gr.inputs.Slider(minimum=320, maximum=1280, default=640, step=32, label="Image Size"),
+    gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.25, step=0.05, label="Confidence Threshold"),
+    gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.7, step=0.05, label="IOU Threshold"),
+]
+
+outputs = [gr.outputs.Image(type="filepath", label="Output Image"), "text"]
+title = "YOLOv8"
+
+examples = [['zidane.jpg', 'yolov8s.pt', 640, 0.25, 0.7], ['bus.jpg', 'yolov8s.pt', 640, 0.25, 0.7]]
+
+demo_app = gr.Interface(
+    fn=yolov8_predictor,
+    inputs=inputs,
+    outputs=outputs,
+    title=title,
+    examples=examples,
+    cache_examples=True,
+    theme='huggingface',
+)
+
+demo_app.launch(debug=True, enable_queue=True)

requirements.txt

@@ -0,0 +1,4 @@
+opencv_python
+torch
+ultralytics==8.0.113
+gradio_client==0.2.7

yolov8n.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:31e20dde3def09e2cf938c7be6fe23d9150bbbe503982af13345706515f2ef95
+size 6534387

yolov8s.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:268e5bb54c640c96c3510224833bc2eeacab4135c6deb41502156e39986b562d
+size 22573363