AnimeHeads-ONNX

AnimeHeads ONNX / アニメ頭部検出 ONNX

This repository contains ONNX exports of nyuuzyou/AnimeHeads anime head detection models. The models detect one class: head.

このリポジトリは nyuuzyou/AnimeHeads のアニメ頭部検出モデルを ONNX に変換したものです。 検出クラスは head の1種類です。

Files / ファイル

• weights/animeheadsv3.onnx - recommended final model, converted from weights/animeheadsv3.pt
• weights/animeheadsv2.onnx - previous version, converted from weights/animeheadsv2.pt
• example.jpg - sample image from the upstream model repository
• inference_opencv_dnn.py - minimal OpenCV DNN inference example
• LICENSE - WTFPL, matching the upstream Hugging Face model card
• THIRD_PARTY_NOTICES.md - upstream and conversion notes

animeheads.pt / v1 was not included as ONNX because it is an older YOLOv5 checkpoint and is not directly forward-compatible with the current Ultralytics exporter used here.

animeheads.pt / v1 は古い YOLOv5 形式で、今回使用した現行 Ultralytics exporter では直接変換できなかったため、ONNX同梱対象から外しています。

Model Details / モデル詳細

• Task: object detection
• Class: head
• Input: 1 x 3 x 640 x 640, RGB, float32, normalized to 0..1
• Output: 1 x 5 x 8400
• Output layout: [cx, cy, w, h, confidence] for one class
• Export settings: ONNX, opset 12, static 640x640 input, no embedded NMS

The exported files were checked with both ONNX Runtime and OpenCV DNN.

ONNX Runtime と OpenCV DNN の両方で読み込みと forward を確認しています。

Usage with OpenCV DNN / OpenCV DNNでの使用

pip install opencv-python numpy
python inference_opencv_dnn.py --model weights/animeheadsv3.onnx --image example.jpg

The script prints detected boxes as JSON.

スクリプトは検出されたボックスをJSONで出力します。

Usage Notes / 利用メモ

For commercial applications, prefer loading these ONNX files directly with OpenCV DNN or ONNX Runtime. This avoids bundling the Ultralytics Python runtime into your application.

商用用途では、このONNXを OpenCV DNN または ONNX Runtime で直接読む構成を推奨します。アプリケーションに Ultralytics Python runtime を同梱せずに済みます。

License / ライセンス

The upstream Hugging Face model card for nyuuzyou/AnimeHeads is marked as wtfpl. The Roboflow Universe page for the related AnimeHeads dataset/model is marked as Public Domain. This repository includes the WTFPL license to match the upstream Hugging Face model metadata.

元の Hugging Face モデルカード nyuuzyou/AnimeHeads は wtfpl 表記です。関連する Roboflow Universe の AnimeHeads データセット/モデルページは Public Domain 表記です。このリポジトリでは Hugging Face 側のモデルメタデータに合わせて WTFPL を同梱しています。

This is not legal advice. If your use case has strict compliance requirements, review the upstream sources and the runtime libraries you distribute.

これは法的助言ではありません。厳密なコンプライアンスが必要な用途では、元配布元と同梱する推論ランタイムのライセンスを確認してください。

Sources / 出典

• Upstream model: https://huggingface.co/nyuuzyou/AnimeHeads
• Related dataset/model page: https://universe.roboflow.com/nyuuzyou/animeheads