init commit

app.py

@@ -0,0 +1,42 @@
+import gradio as gr
+import numpy as np
+import cv2
+from huggingface_hub import hf_hub_download
+from ultralytics import YOLO
+
+checkpoint = hf_hub_download(repo_id="bulkobubulko/cap", filename="yolo11-tapas.pt")
+model = YOLO(checkpoint)
+model.fuse()
+
+def plot_bl(results):
+    img = results[0].orig_img
+    names = results[0].names
+    scores = results[0].boxes.conf.numpy()
+    classes = results[0].boxes.cls.numpy()
+    boxes = results[0].boxes.xyxy.numpy().astype(np.int32)
+
+    num_objects = len(boxes)
+
+    for score, cls, box in zip(scores, classes, boxes):
+        cl_label = names[cls]
+        label = f"{cl_label}: {score:0.2f}"
+
+        img = cv2.rectangle(img.copy(), (box[0], box[1]), (box[2], box[3]), color=(225, 0, 225), thickness=3)
+        img = cv2.putText(img.copy(), label, (box[0], box[1]), fontFace=cv2.FONT_HERSHEY_SIMPLEX, fontScale=1.0, color=(255, 0, 255), thickness=2)
+        img = cv2.putText(img.copy(), f'num caps: {num_objects}', (100, 100), fontFace=cv2.FONT_HERSHEY_SIMPLEX, fontScale=1.0, color=(255, 0, 255), thickness=2)
+    return img 
+
+def predict(frame):
+    frame = model(frame)
+    frame = plot_bl(frame)
+    return frame
+
+with gr.Blocks() as demo:
+    with gr.Row():
+        with gr.Column():
+            input_frame = gr.Image(sources=["webcam"], type="numpy")
+        with gr.Column():
+            output_img = gr.Image(streaming=True)
+        dep = input_frame.stream(predict, [input_frame], [output_img], time_limit=30, stream_every=0.1, concurrency_limit=30)
+
+demo.launch()

requirements.txt

@@ -0,0 +1,100 @@
+aiofiles==23.2.1
+annotated-types==0.7.0
+anyio==4.7.0
+appnope==0.1.4
+asttokens==3.0.0
+certifi==2024.12.14
+charset-normalizer==3.4.1
+click==8.1.8
+comm==0.2.2
+contourpy==1.3.1
+cycler==0.12.1
+debugpy==1.8.11
+decorator==5.1.1
+executing==2.1.0
+fastapi==0.115.6
+ffmpy==0.5.0
+filelock==3.16.1
+fonttools==4.55.3
+fsspec==2024.12.0
+gradio==5.9.1
+gradio_client==1.5.2
+h11==0.14.0
+httpcore==1.0.7
+httpx==0.28.1
+huggingface-hub==0.27.0
+idna==3.10
+ipykernel==6.29.5
+ipython==8.31.0
+jedi==0.19.2
+Jinja2==3.1.5
+jupyter_client==8.6.3
+jupyter_core==5.7.2
+kiwisolver==1.4.7
+lap==0.5.12
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+matplotlib==3.10.0
+matplotlib-inline==0.1.7
+mdurl==0.1.2
+mpmath==1.3.0
+nest-asyncio==1.6.0
+networkx==3.4.2
+numpy==1.26.4
+opencv-python==4.10.0.84
+orjson==3.10.12
+packaging==24.2
+pandas==2.2.3
+parso==0.8.4
+pexpect==4.9.0
+pillow==11.0.0
+platformdirs==4.3.6
+prompt_toolkit==3.0.48
+psutil==6.1.1
+ptyprocess==0.7.0
+pure_eval==0.2.3
+py-cpuinfo==9.0.0
+pydantic==2.10.4
+pydantic_core==2.27.2
+pydub==0.25.1
+Pygments==2.18.0
+pyparsing==3.2.0
+python-dateutil==2.9.0.post0
+python-multipart==0.0.20
+pytz==2024.2
+PyYAML==6.0.2
+pyzmq==26.2.0
+regex==2024.11.6
+requests==2.32.3
+rich==13.9.4
+ruff==0.8.4
+safehttpx==0.1.6
+safetensors==0.4.5
+scipy==1.14.1
+seaborn==0.13.2
+semantic-version==2.10.0
+setuptools==75.6.0
+shapely==2.0.6
+shellingham==1.5.4
+six==1.17.0
+sniffio==1.3.1
+stack-data==0.6.3
+starlette==0.41.3
+sympy==1.13.1
+tokenizers==0.21.0
+tomlkit==0.13.2
+torch==2.5.1
+torchvision==0.20.1
+tornado==6.4.2
+tqdm==4.67.1
+traitlets==5.14.3
+transformers==4.47.1
+typer==0.15.1
+typing_extensions==4.12.2
+tzdata==2024.2
+ultralytics==8.3.55
+ultralytics-thop==2.0.13
+urllib3==2.3.0
+uvicorn==0.34.0
+wcwidth==0.2.13
+websockets==14.1