Overview
I created an inference app using Hugging Face Spaces and the YOLOv5 model (trained on the NDL-DocL dataset) introduced in the following article.
Building a Layout Extraction Model Using the NDL-DocL Dataset and YOLOv5
Building a layout extraction model using the NDL-DocL dataset and YOLOv5
You can try it at the following URL.
Yolov5 Ndl Layout - a Hugging Face Space by nakamura196
Discover amazing ML apps made by the community
You can also download the source code and trained model from the following URL. We hope this serves as a reference when developing similar applications.
Yolov5 Ndl Layout - a Hugging Face Space by nakamura196
Discover amazing ML apps made by the community
The application development referenced the following Space.
YOLOv5 - a Hugging Face Space by pytorch
Discover amazing ML apps made by the community
Usage
You can upload an image or select one from the Examples. The recognition results can be viewed as shown below.
Using the API
I created a notebook on how to use the API. We hope this is also helpful.
https://colab.research.google.com/github/nakamura196/ndl_ocr/blob/main/gradioのAPI利用.ipynb
Comments
…