📌

Hugging Face SpacesとYOLOv5モデル(顔コレデータセットで学習済み)を使った推論アプリの構築

2024/10/05に公開

概要

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を用いたモデルですが、今後より新しいモデルも試してみたいと思います。

Discussion