|
--- |
|
license: mit |
|
pipeline_tag: object-detection |
|
tags: |
|
- YOLO |
|
- Hatsune Miku |
|
--- |
|
|
|
<h1 style="color: #00ffdd; font-size: 2rem;">Dze je Miku? 🥵 (where is Miku)</h1> |
|
|
|
Finetuned [YOLOv8s](https://github.com/ultralytics/ultralytics) on 200 labeled Miku photos for 150 epochs, detects a Hatsune Miku doll in an image 🥰. |
|
|
|
Demo WASM deploy (runs in your browser): https://plasmoxy.github.io/dzejemiku |
|
|
|
```py |
|
!pip install -U "huggingface_hub[cli]" |
|
!pip install ultralytics |
|
!huggingface-cli download Plasmoxy/dze-je-miku-yolo miku_yolo.pt --local-dir . |
|
|
|
from ultralytics import YOLO |
|
|
|
model = YOLO('./miku_yolo.pt') |
|
|
|
res = model.predict([ |
|
'test.jpeg' |
|
], save=True, save_txt=True) |
|
|
|
for r in res: |
|
for b in r.boxes: |
|
print(b.xyxy) |
|
|
|
``` |
|
|
|
<img src="https://cdn-uploads.huggingface.co/production/uploads/6374c3da905caa7e85bbb0c6/AeH2OljyGQ39jZEsixO6W.png" width="500" /> |
|
<img src="https://cdn-uploads.huggingface.co/production/uploads/6374c3da905caa7e85bbb0c6/YLVJWOlbUM-6u5CfLrDJ2.png" width="500" /> |
|
|
|
|
|
|
