概要

Hugging Face Spacesと、YOLOv5モデル(顔コレデータセットで学習済み)を使った推論アプリを作成しました。

人文学オープンデータ共同利用センターが公開する顔コレデータセットは以下です。

Yingtao Tian, Chikahiko Suzuki, Tarin Clanuwat, Mikel Bober-Irizar, Alex Lamb, Asanobu Kitamoto, "KaoKore: A Pre-modern Japanese Art Facial Expression Dataset", arXiv:2002.08595.

http://codh.rois.ac.jp/face/dataset/

推論アプリは、以下のURLからお試しいただけます。

https://huggingface.co/spaces/nakamura196/yolov5-face

また以下のURLからソースコードや学習済みモデルをダウンロードすることができます。同様のアプリケーションを開発される際の参考になれば幸いです。

https://huggingface.co/spaces/nakamura196/yolov5-face/tree/main

なお、アプリケーションの開発には、以下のSpaceを参考にしています。

https://huggingface.co/spaces/pytorch/YOLOv5

使い方

画像をアップロードするか、Examplesから画像を選択してご利用いただけます。以下のように認識結果を確認することができます。

『文正草子』(日本古典籍データセット(国文研所蔵)CODH配信)

まとめ

検出精度が十分でない点がありますが、参考になりましたら幸いです。

現在はYOLOv5を用いたモデルですが、今後より新しいモデルも試してみたいと思います。


動画版(生成AIによる自動生成): この記事の内容をずんだもん×四国めたんの掛け合いで解説しています。自動生成のため、内容に誤りがある可能性があります。正確な情報は記事本文をご参照ください。