ドキュメント内のUsageのコードのエラー報告

[tamoharu]

yomitokuを使わせていただきました！
非常に高精度で素晴らしい結果でした。
開発していただきありがとうございます。

ドキュメントにあるUsageのサンプルコードですが、そのままだといくつかの箇所でエラーになったので報告させていただきます。

元のコード

import cv2

from yomitoku import DocumentAnalyzer
from yomitoku.data.functions import load_image

if __name__ == "__main__":
    img = load_image(PATH_IMAGE)
    analyzer = DocumentAnalyzer(configs=None, visualize=True, device="cuda")
    results, ocr_vis, layout_vis = analyzer(img)

    # HTML形式で解析結果をエクスポート
    results.to_html(PATH_OUTPUT)

    # 可視化画像を保存
    cv2.imwrite("output_ocr.jpg", ocr_vis)
    cv2.imwrite("output_layout.jpg", layout_vis)

• configがNoneのままだとyomitoku/document_analyzer.py", line 179の初期化の際に型チェックでエラーになる
• to_html関数でimg変数を渡さないとyomitoku/export/export_html.py", line 116で型エラーになる

以下のように修正したところ動きました。

import cv2

from yomitoku import DocumentAnalyzer
from yomitoku.data.functions import load_image

if __name__ == "__main__":
    img = load_image(PATH_IMAGE)
    analyzer = DocumentAnalyzer(configs={}, visualize=True, device="cuda")
    results, ocr_vis, layout_vis = analyzer(img)

    # HTML形式で解析結果をエクスポート
    results.to_html(PATH_OUTPUT, img=img)

    # 可視化画像を保存
    cv2.imwrite("output_ocr.jpg", ocr_vis)
    cv2.imwrite("output_layout.jpg", layout_vis)

[kotaro-kinoshita]

@tamoharu
ありがとうございます。エラーに関して認識いたしましたので、修正いたします。