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

Downloads last month: -; Downloads are not tracked for this model. How to track