search

axinc-ai / ailia-models

The collection of pre-trained, state-of-the-art AI models for ailia SDK
2.04k stars 325 forks source link

ADD unstructured-inference #1352

Closed kyakuno closed 8 months ago

kyakuno commented 11 months ago

ドキュメントのレイアウト解析モデル。 https://github.com/Unstructured-IO/unstructured-inference

kyakuno commented 11 months ago

yoloxの方をエクスポートしたい。

from unstructured_inference.models.base import get_model
from unstructured_inference.inference.layout import DocumentLayout

model = get_model("yolox")
layout = DocumentLayout.from_file("sample-docs/layout-parser-paper.pdf", detection_model=model)
Akasan commented 10 months ago

対応方法

  1. yoloxモデルはailia SDK 経由で呼び出す
    • bboxの座標があれば良い
  2. LayoutElementに格納する