Initial YOLOv8 sitting posture upload

README.md

@@ -0,0 +1,15 @@
+
+# YOLOv8 Sitting Posture Detection
+
+This repository contains a YOLOv8 model trained to detect sitting postures.
+
+## How to Use
+
+### Locally
+1. Install dependencies:
+```bash
+pip install -r requirements.txt
+
+2. Run the Gradio app
+```bash
+python app.py

app.py

@@ -0,0 +1,22 @@
+
+import gradio as gr
+from inference import predict
+from PIL import Image
+import io
+
+def predict_image(img: Image.Image):
+    with io.BytesIO() as buffer:
+        img.save(buffer, format="JPEG")
+        img_bytes = buffer.getvalue()
+    return predict(img_bytes)
+
+iface = gr.Interface(
+    fn=predict_image,
+    inputs=gr.Image(type="pil"),
+    outputs=gr.JSON(),
+    title="Sitting Posture Detection",
+    description="Upload an image to detect sitting postures using YOLOv8"
+)
+
+if __name__ == "__main__":
+    iface.launch()

inference.py

@@ -0,0 +1,21 @@
+
+import torch
+from ultralytics import YOLO
+from PIL import Image
+import io
+
+# Load YOLOv8 model
+model = YOLO("model.pt")
+
+def predict(image_bytes):
+    img = Image.open(io.BytesIO(image_bytes))
+    results = model.predict(img)
+    output = []
+    for result in results:
+        for i in range(len(result.boxes)):
+            output.append({
+                "bbox": result.boxes.xyxy[i].tolist(),
+                "class": int(result.boxes.cls[i].item()),
+                "confidence": float(result.boxes.conf[i].item())
+            })
+    return output

model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:65d11808479be328cbad89a7a8a46114886a45b1e57782de8e115673ab05a453
+size 23121944

requirements.txt

@@ -0,0 +1,6 @@
+
+ultralytics>=8.0
+torch>=2.0
+Pillow
+gradio
+huggingface_hub