Yolov11n Face Detection

🚀 YOLOv11n-Face-Detection

この軽量な顔検出モデルは、YOLOアーキテクチャ（YOLOv11 nano）に基づいており、WIDERFACEデータセットで225エポックの学習を行っています。このモデルは、顔検出タスクに特化しており、様々なシーンでの顔検出に役立ちます。

評価セットでは、以下の結果を達成しています。

==================== Results ====================
Easy   Val AP: 0.9420471677096086
Medium Val AP: 0.9210357271019756
Hard   Val AP: 0.8099848364072022
=================================================

YOLOの結果:

混同行列:

[[23577 2878]

[16098 0]]

✨ 主な機能

• YOLOアーキテクチャをベースにした軽量な顔検出モデル。
• WIDERFACEデータセットで学習され、高い精度を達成。

📦 インストール

このモデルを使用するには、huggingface_hubとultralyticsライブラリが必要です。以下のコマンドでインストールできます。

pip install huggingface_hub ultralytics

💻 使用例

基本的な使用法

from huggingface_hub import hf_hub_download
from ultralytics import YOLO

model_path = hf_hub_download(repo_id="AdamCodd/YOLOv11n-face-detection", filename="model.pt")
model = YOLO(model_path)

results = model.predict("/path/to/your/image", save=True) # 結果を runs/detect/predict に保存

📚 ドキュメント

制限事項

• 極端な照明条件では性能が変動する可能性があります。
• 正面およびやや斜めの顔に最適です。
• 顔が20ピクセル以上を占める場合に最適な性能を発揮します。

📄 ライセンス

このモデルは、Apache-2.0ライセンスの下で提供されています。

モデル情報