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