🙆

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

2022/08/04に公開

概要

Hugging Face Spacesと、以下の記事で紹介したYOLOv5モデル(NDL-DocLデータセットで学習済み)を使った推論アプリを作成しました。

https://zenn.dev/nakamura196/articles/5f09c992e10694

以下のURLからお試しいただけます。

https://huggingface.co/spaces/nakamura196/yolov5-ndl-layout

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

https://huggingface.co/spaces/nakamura196/yolov5-ndl-layout

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

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

使い方

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

APIの利用

APIの利用方法に関するノートブックを作成しました。こちらも参考になりましたら幸いです。

https://colab.research.google.com/github/nakamura196/ndl_ocr/blob/main/gradioのAPI利用.ipynb

Discussion