Upload folder using huggingface_hub

.gitignore

@@ -0,0 +1,8 @@
+yolov8l.pt
+yolov8n-cls.pt
+yolov8n-pose.pt
+yolov8n-seg.pt
+
+_pychache_
+
+.venv/

README.md

@@ -1,12 +1,25 @@
 ---
-title: YOLOv8 WebUI
-emoji: 🐠
-colorFrom: gray
-colorTo: blue
+title: YOLOv8-WebUI
+app_file: app.py
 sdk: gradio
 sdk_version: 3.35.2
-app_file: app.py
-pinned: false
 ---
+<h1 align="center">YOLOv8-WebUI</h1>
+<div align="center">
+
+Web browser UI for [YOLOv8](https://github.com/ultralytics/ultralytics)
+
+</div>
+
+
+# Launch
+
+## Windows
+1.Clone this repository.
+
+2.Run `webui.bat`  
+
+## Linux or Mac
+1.Clone this repository.
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+2.Run `webui.sh`  

app.py

@@ -0,0 +1,80 @@
+import gradio as gr
+from ultralytics import YOLO
+
+#各モデルをロード
+detect_model = YOLO("best.pt")
+seg_model = YOLO('yolov8n-seg.pt')
+cls_model = YOLO('yolov8n-cls.pt')
+pose_model = YOLO("yolov8n-pose.pt")
+
+#オプションリストをオブジェクトに整形
+def return_options(checkbox):
+    option = {}
+
+    #整形
+    for check in checkbox:
+        option[check] = True
+
+    return option
+
+#結果を描画する
+def plot(res):
+    plot = res[0].plot()
+    return plot
+
+def inference(type,input,conf):  
+    if(type == "Detección"):
+        model = detect_model
+    elif(type == "seg"):
+        model = seg_model
+    elif(type == "cls"):
+        model = cls_model
+    elif(type == "Pose"):
+        model = pose_model
+
+    cpu=True
+    if(cpu):
+        device = "cpu"
+
+    
+    line_width = None
+
+    
+    
+    #物体検出を実行
+    res = model(input,conf=conf,iou=0.7,device="cpu",max_det=300,line_width=line_width)
+
+    #結果を描画
+    plotted = plot(res)
+    return plotted
+
+with gr.Blocks() as app:
+    #ヘッダー
+    gr.Markdown("# Detección de comportamiento sospechoso")
+
+    #タブ
+    with gr.Tabs():
+        #inferenceタブ
+        with gr.TabItem("Inferencia"):
+            with gr.Row():
+                input = gr.Image()
+                output = gr.Image()
+
+            type = gr.Radio(["Detección", "Pose"], value="Detección", label="Tasks")
+
+            #オプション 
+            conf = gr.Slider(minimum=0, maximum=1, value=0.25, step=0.01, interactive=True,label="conf")
+            #iou = gr.Slider(minimum=0, maximum=1, value=0.7, step=0.01, interactive=True,label="iou")
+            #checkbox = gr.CheckboxGroup(["half","show","save","save_txt","save_conf","save_crop","hide_labels","hide_conf","vid_stride","visualize","augment","agnostic_nms","retina_masks","boxes"], label="Options",value=["boxes"])
+
+            #device = gr.Number(value=0, label="device", interactive=True, precision=0)
+            #cpu = gr.Checkbox(label="cpu", interactive=True)
+            #max_det = gr.Number(value=300, label="max_det", interactive=True, precision=0)
+            #line_width = gr.Number(value=0, label="line_width", interactive=True, precision=0)
+
+            inference_button = gr.Button("Realizar Inferencia")
+
+
+
+    #inputから画像を取得してdetect関数を実行
+    inference_button.click(inference, inputs=[type,input,conf], outputs=output)

best.pt

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

launch.py

@@ -0,0 +1,34 @@
+import importlib.util
+import pip
+import app 
+import sys
+
+#パッケージがインストールされているかの確認
+def is_installed(package):
+    try:
+        spec = importlib.util.find_spec(package)
+    except ModuleNotFoundError:
+        return False
+
+    return spec is not None
+
+def start():
+    print(f"Python {sys.version}")
+
+    print("###################################################################")
+
+    #パッケージがインストールされていなければパッケージをインストールする
+    if not (is_installed("gradio")):
+
+        print("Installing gradio")
+        pip.main(['install','gradio'])
+
+    if not (is_installed("ultralytics")):
+
+        print("Installing ultralytics")
+        pip.main(['install','ultralytics'])
+
+    #appを起動
+    app.app.launch(share=True)
+
+start()

requirements.txt

@@ -0,0 +1,2 @@
+gradio
+ultralytics

webui.bat

@@ -0,0 +1,6 @@
+@echo off
+cd /d %~dp0    
+
+echo Launching app...
+
+python launch.py

webui.sh

@@ -0,0 +1,5 @@
+#!bin/bash
+
+echo Launching app...
+
+python launch.py

yolov8l.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:18218ea4798da042d9862e6029ca9531adbd40ace19b6c9a75e2e28f1adf30cc
+size 87769683

yolov8n-cls.pt

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

yolov8n-pose.pt

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

yolov8n-seg.pt

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