CV / YOLOv6

YOLOv6

reComputer R Series (CM5 + Hailo-10H)

YOLOv6 COCO-Objekterkennung auf reComputer RK3576/RK3588 (RKNN) und reComputer R Series mit Hailo-8 oder Hailo-10H.

97 Downloads
Präzision
RKNN / Hailo HEF

Wähle das Gerät, das du verwendest. Die Einrichtungsanleitung und Dokumentation werden entsprechend aktualisiert.

Erste Schritte

Bereitstellen
sudo docker run --rm \
  --name cm5-hailo10h-yolov6n \
  --privileged \
  --net=host \
  -e PYTHONUNBUFFERED=1 \
  --device /dev/hailo0:/dev/hailo0 \
  -v /usr/lib/libhailort.so.5.1.1:/usr/lib/libhailort.so.5.1.1:ro \
  -v /usr/lib/libhailort.so:/usr/lib/libhailort.so:ro \
  ghcr.io/seeed-projects/recomputer-hailo10h-cv/yolov6n:latest \
  python web_detection.py --model_path model/yolov6n.hef --video_path video/test.mp4

REST API

Verwende die REST API für die Inferenz. Kopiere die folgenden Befehle.

Curl
curl http://localhost:8080/v1/chat/completions -d '{
  "model": "yolov6",
  "messages": [{"role": "user", "content": "Hello"}]
}'
Python
import requests

resp = requests.post(
    "http://localhost:8080/v1/chat/completions",
    json={"model": "yolov6", "messages": [{"role": "user", "content": "Hello"}]},
)
print(resp.json())

Modelldetails

reComputer R Series (CM5 + Hailo-10H)

YOLOv6n on reComputer R Series (CM5 + Hailo-10H)

YOLOv6n führt COCO-Objekterkennung (80 Klassen) auf Hailo-10H über HailoRT aus. Das kompilierte HEF liefert entweder das On-Chip-HPP-NMS-Ergebnis oder die neun rohen Split-Heads; die Anwendung verarbeitet beide und protokolliert, welches Layout das HEF geliefert hat.

Diese Seite richtet sich an reComputer R Series (CM5 + Hailo-10H) mit einem PCIe-Hailo-10H-Beschleuniger.

Hardware und Host-Setup

ElementWert
BoardreComputer R Series mit Raspberry Pi CM5
BeschleunigerHailo-10H über PCIe, verfügbar als /dev/hailo0
RuntimeHailoRT 5.1.1
Python im Container3.13, aarch64

Installieren Sie die Hailo-10H-Pakete auf dem Host und prüfen Sie Treiber und Gerät, bevor Sie den Container starten:

bash
sudo apt update
sudo apt install hailo-h10-all -y
sudo reboot

# After reboot
hailortcli --version
hailortcli fw-control identify
ls -l /dev/hailo0

Die erwartete HailoRT-Version für dieses Image ist 5.1.1.

Ausführung mit dem Demo-Video

bash
sudo docker run --rm \
  --name cm5-hailo10h-yolov6n \
  --privileged \
  --net=host \
  -e PYTHONUNBUFFERED=1 \
  --device /dev/hailo0:/dev/hailo0 \
  -v /usr/lib/libhailort.so.5.1.1:/usr/lib/libhailort.so.5.1.1:ro \
  -v /usr/lib/libhailort.so:/usr/lib/libhailort.so:ro \
  ghcr.io/seeed-projects/recomputer-hailo10h-cv/yolov6n:latest \
  python web_detection.py --model_path model/yolov6n.hef --video_path video/test.mp4

Öffnen Sie http://<Board_IP>:8000 für die Webvorschau.

USB-Kameramodus

bash
sudo docker run --rm \
  --name cm5-hailo10h-yolov6n \
  --privileged \
  --net=host \
  -e PYTHONUNBUFFERED=1 \
  --device /dev/hailo0:/dev/hailo0 \
  --device /dev/video0:/dev/video0 \
  -v /usr/lib/libhailort.so.5.1.1:/usr/lib/libhailort.so.5.1.1:ro \
  -v /usr/lib/libhailort.so:/usr/lib/libhailort.so:ro \
  ghcr.io/seeed-projects/recomputer-hailo10h-cv/yolov6n:latest \
  python web_detection.py --model_path model/yolov6n.hef --camera_id 0

HEF-Datei

HEFGrößeHinweise
yolov6n.hef3,862,528 (3.7 MB)Das HEF stammt aus dem v5.4.0-Build des Hailo Model Zoo für Hailo-10H; Parameter 4.32M, Operationen 11.12G.

Source: https://hailo-model-zoo.s3.eu-west-2.amazonaws.com/ModelZoo/Compiled/v5.4.0/hailo10h/yolov6n.hef.

REST API

text
POST http://<Board_IP>:8000/api/models/yolov6n/predict
bash
curl -X POST "http://<Board_IP>:8000/api/models/yolov6n/predict" \
  -F "file=@test.jpg"
EndpunktMethodeZweck
/GETWeb-Vorschau-Oberfläche
/api/models/yolov6n/predictPOSTDetektionen (JSON)
/api/video_feedGETMJPEG-Vorschaustream

Implementierungshinweise

  • Der Model Zoo base/yolov6.yaml nutzt hpp=true, meta_arch=yolo_v6, score_threshold=0.03 und nms_iou_thresh=0.65; info.output_shape dokumentiert die neun rohen Split-Heads (4 Box, 1 Objectness, 80 Klassen je Stride 32/16/8).
  • Das Modul verarbeitet beide Compile-Layouts: hat das HEF das NMS bereits ausgeführt (HPP), parst es den Post-NMS-Tensor (kompakter Puffer je Klasse, dichte Formen Cx5xD / CxDx5, ragged NMS-by-score-Liste); liefert es die rohen Split-Heads, dekodiert es die Abstände um den Zellmittelpunkt (Stride-Einheiten), multipliziert sigmoid(Klassen) mit sigmoid(Objectness) und führt ein NMS je Klasse mit dem IOU-Reglerwert aus.
  • [YOLOv6n] layout=... wird bei der ersten Inferenz einmal protokolliert, um das Layout auf der Hardware zu bestätigen.
  • Das HEF enthält normalize_in_net (mean 0 / std 255) und padding_color=114: die Anwendung letterboxt mit Grau (114) und übergibt rohe uint8-RGB-Pixel.
  • Klassen-IDs 0..79 indizieren direkt die standardmäßige COCO-Liste (die Model-Zoo-Evaluierung nutzt labels_offset=1).
  • Standardwerte: Konfidenz 0.25; beim On-Chip-NMS ist der IOU-Regler ohne Wirkung, bei den rohen Heads steuert er das Host-NMS.

Entwicklungshinweise

  • Quellmodul: src/hailo10h_yolov6n/ (Hailo-10H) oder src/rpi5_hailo8_yolov6n/ (Hailo-8) im passenden Modell-Repository
  • Container: ghcr.io/seeed-projects/recomputer-hailo10h-cv/yolov6n:latest
  • Der Hailo Model Zoo veröffentlicht nur YOLOv6n; YOLOv6s/m/l haben kein Hailo-HEF.
  • Modelllizenz: GPL-3.0 (Upstream meituan/YOLOv6).

Eingaben und Ausgaben

Input: image, video frame, or camera frame. Output: COCO classes, confidence scores, and boxes.