yomitoku 0.4.0.post1.dev0__tar.gz

yomitoku-0.4.0.post1.dev0/.github/release-drafter.yml

@@ -0,0 +1,62 @@
+---
+  name-template: "v$RESOLVED_VERSION"
+  tag-template: "v$RESOLVED_VERSION"
+  categories: # categorize
+    - title: "🚀 機能追加"
+      labels:
+        - "enhancement"
+    - title: "🔧  リファクタ"
+      labels:
+        - "refactoring"
+    - title: "🐛 バグ修正"
+      labels:
+        - "bug"
+    - title: "✅ テスト"
+      labels:
+        - "test"
+    - title: "📖 ドキュメント"
+      labels:
+        - "documentation"
+  change-template: "- $TITLE @$AUTHOR (#$NUMBER)"
+  change-title-escapes: '\<*_&' # You can add # and @ to disable mentions, and add ` to disable code blocks.
+  autolabeler: # auto add labels based on branches or titles
+    - label: "enhancement"
+      branch:
+        - '/feature\/.+/'
+        - '/feat\/.+/'
+    - label: "release"
+      branch:
+        - '/release\/.+/'
+    - label: "refactoring"
+      branch:
+        - '/refactor\/.+/'
+      title:
+        - "/refactor/i"
+    - label: "bug"
+      branch:
+        - '/fix\/.+/'
+        - '/bug\/.+/'
+      title:
+        - "/fix/i"
+        - "/bug/i"
+    - label: "test"
+      branch:
+        - '/test\/.+/'
+    - label: "documentation"
+      branch:
+        - '/doc\/.+/'
+      title:
+        - "/doc/i"
+  version-resolver: # resolve next version based on tags ($RESOLVED_VERSION)
+    major:
+      labels:
+        - "breaking"
+    minor:
+      labels:
+        - "enhancement"
+    default: patch
+  template: |
+    ## 変更
+  
+    $CHANGES
+  

yomitoku-0.4.0.post1.dev0/.github/workflows/build-and-publish-docs.yaml

@@ -0,0 +1,18 @@
+name: ci
+on:
+  push:
+    branches:
+      - main
+permissions:
+  contents: write
+jobs:
+  deploy-docs:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v3
+      - uses: actions/setup-python@v4
+        with:
+          python-version: 3.x
+      - run: | 
+          pip install tox
+          tox -e docs

yomitoku-0.4.0.post1.dev0/.github/workflows/build-and-publish.yml

@@ -0,0 +1,28 @@
+name: Publish to PyPI
+
+on:
+  release:
+    types: [published]
+
+jobs:
+  publish_PyPI:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v2
+        with:
+          fetch-depth: 0
+          tags: true
+      - name: Set up Python 3.9
+        uses: actions/setup-python@v3
+        with:
+          python-version: "3.9"
+      - name: Install uv
+        uses: astral-sh/setup-uv@v3
+        with:
+          enable-cache: true
+      - name: Install dependencies
+        run: uv sync --dev
+      - name: build
+        run: uv build
+      - name: Publish to PyPI
+        run: uv publish --token ${{ secrets.PYPI }}

yomitoku-0.4.0.post1.dev0/.github/workflows/create-release.yml

@@ -0,0 +1,23 @@
+---
+name: Release Drafter
+
+on:
+  push:
+    branches:
+      - main
+  pull_request_target: 
+    types: [opened, reopened, synchronize]
+
+permissions:
+  contents: read
+
+jobs:
+  update_release_draft:
+    permissions:
+      contents: write
+      pull-requests: write
+    runs-on: ubuntu-latest
+    steps:
+      - uses: release-drafter/release-drafter@v6
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}

yomitoku-0.4.0.post1.dev0/.github/workflows/lint-and-test.yml

@@ -0,0 +1,40 @@
+name: Python Lint
+
+on:
+  pull_request:
+    branches:
+      - main
+
+permissions:
+  contents: read
+
+jobs:
+  lint-and-test:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - name: Cleanup disk space
+        run: |
+          sudo apt-get clean
+          sudo rm -rf /usr/share/dotnet
+          sudo rm -rf /usr/local/lib/android
+          sudo rm -rf /opt/ghc
+          df -h
+      - name: Set up Python 3.12
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+      - name: Install uv
+        uses: astral-sh/setup-uv@v3
+        with:
+          enable-cache: true
+      - name: pin python version
+        run: uv python pin 3.12
+      - name: Install dependencies
+        run: sudo apt install -y libopencv-dev poppler-utils
+      - name: Install tox-uv
+        run: uv tool install tox --with tox-uv
+      - name: Run linter
+        run: tox -e lint
+      - name: Run tests
+        run:  tox -p -e py39,py310,py311,py312

yomitoku-0.4.0.post1.dev0/.gitignore

@@ -0,0 +1,23 @@
+# Python-generated files
+__pycache__/
+*.py[oc]
+build/
+dist/
+wheels/
+*.egg-info
+
+# Virtual environments
+.venv
+.tox
+
+# Dev tools cache
+.ruff_cache
+.pytest_cache
+
+.DS_Store
+
+dataset/
+weights/
+results/
+
+.coverage*

yomitoku-0.4.0.post1.dev0/.pre-commit-config.yaml

@@ -0,0 +1,13 @@
+repos:
+  - repo: local
+    hooks:
+      - id: lint
+        name: lint
+        entry: bash -c '.tox/lint/bin/ruff check --fix'
+        language: system
+        types: [python]
+      - id: format
+        name: format
+        entry: bash -c '.tox/lint/bin/ruff format'
+        language: system
+        types: [python]

yomitoku-0.4.0.post1.dev0/.python-version

@@ -0,0 +1 @@
+3.9

yomitoku-0.4.0.post1.dev0/PKG-INFO

@@ -0,0 +1,127 @@
+Metadata-Version: 2.3
+Name: yomitoku
+Version: 0.4.0.post1.dev0
+Summary: Yomitoku is a document image analysis package powered by AI technology for the Japanese language.
+Author-email: Kotaro Kinoshita <kotaro.kinoshita@mlism.com>
+License: CC BY-NC-SA 4.0
+Keywords: Deep Learning,Japanese,OCR
+Requires-Python: >=3.9
+Requires-Dist: huggingface-hub>=0.26.1
+Requires-Dist: lxml>=5.3.0
+Requires-Dist: omegaconf>=2.3.0
+Requires-Dist: opencv-python>=4.10.0.84
+Requires-Dist: pdf2image>=1.17.0
+Requires-Dist: pyclipper>=1.3.0.post6
+Requires-Dist: pydantic>=2.9.2
+Requires-Dist: shapely>=2.0.6
+Requires-Dist: timm>=1.0.11
+Requires-Dist: torch>=2.5.0
+Requires-Dist: torchvision>=0.20.0
+Description-Content-Type: text/markdown
+
+# YomiToku
+
+![Python](https://img.shields.io/badge/Python-3.9|3.10|3.11|3.12-F9DC3E.svg?logo=python&logoColor=&style=flat)
+![Pytorch](https://img.shields.io/badge/Pytorch-2.5-EE4C2C.svg?logo=Pytorch&style=fla)
+![OS](https://img.shields.io/badge/OS-Linux|MacOS-1793D1.svg?&style=fla)
+[![Document](https://img.shields.io/badge/docs-live-brightgreen)](https://kotaro-kinoshita.github.io/yomitoku-dev/)
+
+<img src="static/logo/horizontal.png" width="800px">
+
+## 🌟 概要
+
+YomiToku は日本語に特化した AI 文章画像解析エンジン(Document AI)です。画像内の文字の全文 OCR およびレイアウト解析機能を有しており、画像内の文字情報や図表を認識、抽出、変換します。
+
+- 🤖 日本語データセットで学習した 4 種類(文字位置の検知、文字列認識、レイアウト解析、表の構造認識)の AI モデルを搭載しています。4 種類のモデルはすべて独自に学習されたモデルで日本語文書に対して、高精度に推論可能です。
+- 🇯🇵 各モデルは日本語の文書画像に特化して学習されており、7000 文字を超える日本語文字の認識をサーポート、縦書きなど日本語特有のレイアウト構造の文書画像の解析も可能です。（日本語以外にも英語の文書に対しても対応しています）。
+- 📈 レイアウト解析、表の構造解析, 読み順推定機能により、文書画像のレイアウトの意味的構造を壊さずに情報を抽出することが可能です。
+- 📄 多様な出力形式をサポートしています。html やマークダウン、json、csv のいずれかのフォーマットに変換可能です。また、文書内に含まれる図表、画像の抽出の出力も可能です。
+- ⚡ GPU 環境で高速に動作し、効率的に文書の文字起こし解析が可能です。また、VRAM も 8GB 以内で動作し、ハイエンドな GPU を用意する必要はありません。
+
+## 🖼️ デモ
+
+[gallery.md](gallery.md)にも複数種類の画像の検証結果を掲載しています。
+
+|                          入力画像                          |                       OCR の結果                        |
+| :--------------------------------------------------------: | :-----------------------------------------------------: |
+|        <img src="static/in/demo.jpg" width="400px">        | <img src="static/out/in_demo_p1_ocr.jpg" width="400px"> |
+|                    レイアウト解析の結果                    |     エクスポート<br>(HTML で出力したものをスクショ)     |
+| <img src="static/out/in_demo_p1_layout.jpg" width="400px"> |   <img src="static/out/demo_html.png" width="400px">    |
+
+Markdown でエクスポートした結果は関してはリポジトリ内の[static/out/in_demo_p1.md](static/out/in_demo_p1.md)を参照
+
+- `赤枠` : 図、画像等の位置
+- `緑枠` : 表領域全体の位置
+- `ピンク枠` : 表のセル構造(セル上の文字は [行番号, 列番号] (rowspan x colspan)を表します)
+- `青枠` : 段落、テキストグループ領域
+- `赤矢印` : 読み順推定の結果
+
+画像の出典:[「令和 6 年版情報通信白書 3 章 2 節 AI の進化に伴い発展するテクノロジー」](https://www.soumu.go.jp/johotsusintokei/whitepaper/ja/r06/pdf/n1410000.pdf)：（総務省） を加工して作成
+
+## 📣 リリース情報
+
+- 2024 年 12 月 XX YomiToku vX.X.X を公開
+
+## 💡 インストールの方法
+
+```
+pip install git+https://github.com/kotaro-kinoshita/yomitoku-dev.git@main
+```
+
+- pytorch がご自身の GPU の環境にあったものをインストールしてください
+
+### 依存ライブラリ
+
+pdf ファイルの解析を行うためには、別途、[poppler](https://poppler.freedesktop.org/)のインストールが必要です。
+
+**Mac**
+
+```
+brew install poppler
+```
+
+**Linux**
+
+```
+apt install poppler-utils -y
+```
+
+## 🚀 実行方法
+
+```
+yomitoku ${path_data} -f md -o results -v --figure
+```
+
+- `${path_data}` 解析対象の画像が含まれたディレクトリか画像ファイルのパスを直接して指定してください。ディレクトリを対象とした場合はディレクトリのサブディレクトリ内の画像も含めて処理を実行します。
+- `-f`, `--format` 出力形式のファイルフォーマットを指定します。(json, csv, html, md をサポート)
+- `-o`, `--outdir` 出力先のディレクトリ名を指定します。存在しない場合は新規で作成されます。
+- `-v`, `--vis` を指定すると解析結果を可視化した画像を出力します。
+- `-d`, `--device` モデルを実行するためのデバイスを指定します。gpu が利用できない場合は cpu で推論が実行されます。(デフォルト: cuda)
+- `--ignore_line_break` 画像の改行位置を無視して、段落内の文章を連結して返します。（デフォルト：画像通りの改行位置位置で改行します。）
+- `figure_letter` 検出した図表に含まれる文字も出力ファイルにエクスポートします。
+- `figure` 検出した図、画像を出力ファイルにエクスポートします。(html と markdown のみ)
+
+その他のオプションに関しては、ヘルプを参照
+
+```
+yomitoku --help
+```
+
+### Note
+
+- CPU を用いての推論向けに最適化されておらず、処理時間が長くなりますので、GPU での実行を推奨します。
+- 活字のみ識別をサポートしております。手書き文字に関しては、読み取れる場合もありますが、公式にはサポートしておりません。
+- OCR は文書 OCR と情景 OCR(看板など紙以外にプリントされた文字)に大別されますが、Yomitoku は文書 OCR 向けに最適化されています。
+- AI-OCR の識別精度を高めるために、入力画像の解像度が重要です。低解像度画像では識別精度が低下します。最低でも画像の短辺を 720px 以上の画像で推論することをお勧めします。
+
+## 📝 ドキュメント
+
+パッケージの詳細は[ドキュメント](https://kotaro-kinoshita.github.io/yomitoku-dev/)を確認してください。
+
+## LICENSE
+
+本リポジトリ内に格納されているリソースのライセンスは YomiToku は CC BY-NC-SA 4.0 に従います。
+非商用での個人利用、研究目的での利用は自由に利用できます。
+商用目的での利用に関しては、別途、商用ライセンスを提供しますので、開発者にお問い合わせください。
+
+YomiToku © 2024 by MLism Inc. is licensed under CC BY-NC-SA 4.0. To view a copy of this license, visit https://creativecommons.org/licenses/by-nc-sa/4.0/

yomitoku-0.4.0.post1.dev0/README.md

@@ -0,0 +1,106 @@
+# YomiToku
+
+![Python](https://img.shields.io/badge/Python-3.9|3.10|3.11|3.12-F9DC3E.svg?logo=python&logoColor=&style=flat)
+![Pytorch](https://img.shields.io/badge/Pytorch-2.5-EE4C2C.svg?logo=Pytorch&style=fla)
+![OS](https://img.shields.io/badge/OS-Linux|MacOS-1793D1.svg?&style=fla)
+[![Document](https://img.shields.io/badge/docs-live-brightgreen)](https://kotaro-kinoshita.github.io/yomitoku-dev/)
+
+<img src="static/logo/horizontal.png" width="800px">
+
+## 🌟 概要
+
+YomiToku は日本語に特化した AI 文章画像解析エンジン(Document AI)です。画像内の文字の全文 OCR およびレイアウト解析機能を有しており、画像内の文字情報や図表を認識、抽出、変換します。
+
+- 🤖 日本語データセットで学習した 4 種類(文字位置の検知、文字列認識、レイアウト解析、表の構造認識)の AI モデルを搭載しています。4 種類のモデルはすべて独自に学習されたモデルで日本語文書に対して、高精度に推論可能です。
+- 🇯🇵 各モデルは日本語の文書画像に特化して学習されており、7000 文字を超える日本語文字の認識をサーポート、縦書きなど日本語特有のレイアウト構造の文書画像の解析も可能です。（日本語以外にも英語の文書に対しても対応しています）。
+- 📈 レイアウト解析、表の構造解析, 読み順推定機能により、文書画像のレイアウトの意味的構造を壊さずに情報を抽出することが可能です。
+- 📄 多様な出力形式をサポートしています。html やマークダウン、json、csv のいずれかのフォーマットに変換可能です。また、文書内に含まれる図表、画像の抽出の出力も可能です。
+- ⚡ GPU 環境で高速に動作し、効率的に文書の文字起こし解析が可能です。また、VRAM も 8GB 以内で動作し、ハイエンドな GPU を用意する必要はありません。
+
+## 🖼️ デモ
+
+[gallery.md](gallery.md)にも複数種類の画像の検証結果を掲載しています。
+
+|                          入力画像                          |                       OCR の結果                        |
+| :--------------------------------------------------------: | :-----------------------------------------------------: |
+|        <img src="static/in/demo.jpg" width="400px">        | <img src="static/out/in_demo_p1_ocr.jpg" width="400px"> |
+|                    レイアウト解析の結果                    |     エクスポート<br>(HTML で出力したものをスクショ)     |
+| <img src="static/out/in_demo_p1_layout.jpg" width="400px"> |   <img src="static/out/demo_html.png" width="400px">    |
+
+Markdown でエクスポートした結果は関してはリポジトリ内の[static/out/in_demo_p1.md](static/out/in_demo_p1.md)を参照
+
+- `赤枠` : 図、画像等の位置
+- `緑枠` : 表領域全体の位置
+- `ピンク枠` : 表のセル構造(セル上の文字は [行番号, 列番号] (rowspan x colspan)を表します)
+- `青枠` : 段落、テキストグループ領域
+- `赤矢印` : 読み順推定の結果
+
+画像の出典:[「令和 6 年版情報通信白書 3 章 2 節 AI の進化に伴い発展するテクノロジー」](https://www.soumu.go.jp/johotsusintokei/whitepaper/ja/r06/pdf/n1410000.pdf)：（総務省） を加工して作成
+
+## 📣 リリース情報
+
+- 2024 年 12 月 XX YomiToku vX.X.X を公開
+
+## 💡 インストールの方法
+
+```
+pip install git+https://github.com/kotaro-kinoshita/yomitoku-dev.git@main
+```
+
+- pytorch がご自身の GPU の環境にあったものをインストールしてください
+
+### 依存ライブラリ
+
+pdf ファイルの解析を行うためには、別途、[poppler](https://poppler.freedesktop.org/)のインストールが必要です。
+
+**Mac**
+
+```
+brew install poppler
+```
+
+**Linux**
+
+```
+apt install poppler-utils -y
+```
+
+## 🚀 実行方法
+
+```
+yomitoku ${path_data} -f md -o results -v --figure
+```
+
+- `${path_data}` 解析対象の画像が含まれたディレクトリか画像ファイルのパスを直接して指定してください。ディレクトリを対象とした場合はディレクトリのサブディレクトリ内の画像も含めて処理を実行します。
+- `-f`, `--format` 出力形式のファイルフォーマットを指定します。(json, csv, html, md をサポート)
+- `-o`, `--outdir` 出力先のディレクトリ名を指定します。存在しない場合は新規で作成されます。
+- `-v`, `--vis` を指定すると解析結果を可視化した画像を出力します。
+- `-d`, `--device` モデルを実行するためのデバイスを指定します。gpu が利用できない場合は cpu で推論が実行されます。(デフォルト: cuda)
+- `--ignore_line_break` 画像の改行位置を無視して、段落内の文章を連結して返します。（デフォルト：画像通りの改行位置位置で改行します。）
+- `figure_letter` 検出した図表に含まれる文字も出力ファイルにエクスポートします。
+- `figure` 検出した図、画像を出力ファイルにエクスポートします。(html と markdown のみ)
+
+その他のオプションに関しては、ヘルプを参照
+
+```
+yomitoku --help
+```
+
+### Note
+
+- CPU を用いての推論向けに最適化されておらず、処理時間が長くなりますので、GPU での実行を推奨します。
+- 活字のみ識別をサポートしております。手書き文字に関しては、読み取れる場合もありますが、公式にはサポートしておりません。
+- OCR は文書 OCR と情景 OCR(看板など紙以外にプリントされた文字)に大別されますが、Yomitoku は文書 OCR 向けに最適化されています。
+- AI-OCR の識別精度を高めるために、入力画像の解像度が重要です。低解像度画像では識別精度が低下します。最低でも画像の短辺を 720px 以上の画像で推論することをお勧めします。
+
+## 📝 ドキュメント
+
+パッケージの詳細は[ドキュメント](https://kotaro-kinoshita.github.io/yomitoku-dev/)を確認してください。
+
+## LICENSE
+
+本リポジトリ内に格納されているリソースのライセンスは YomiToku は CC BY-NC-SA 4.0 に従います。
+非商用での個人利用、研究目的での利用は自由に利用できます。
+商用目的での利用に関しては、別途、商用ライセンスを提供しますので、開発者にお問い合わせください。
+
+YomiToku © 2024 by MLism Inc. is licensed under CC BY-NC-SA 4.0. To view a copy of this license, visit https://creativecommons.org/licenses/by-nc-sa/4.0/

yomitoku-0.4.0.post1.dev0/configs/layout_parser_rtdetrv2.yaml

@@ -0,0 +1,67 @@
+hf_hub_repo: KotaroKinoshita/yomitoku-layout-parser-rtdtrv2-open-beta
+thresh_score: 0.45
+data:
+  img_size:
+    - 640
+    - 640
+PResNet:
+  depth: 50
+  variant: d
+  freeze_at: 0
+  return_idx:
+    - 1
+    - 2
+    - 3
+  num_stages: 4
+  freeze_norm: true
+HybridEncoder:
+  in_channels:
+    - 512
+    - 1024
+    - 2048
+  feat_strides:
+    - 8
+    - 16
+    - 32
+  hidden_dim: 256
+  use_encoder_idx:
+    - 2
+  num_encoder_layers: 1
+  nhead: 8
+  dim_feedforward: 1024
+  dropout: 0.0
+  enc_act: gelu
+  expansion: 1.0
+  depth_mult: 1
+  act: silu
+RTDETRTransformerv2:
+  num_classes: 3
+  feat_channels:
+    - 256
+    - 256
+    - 256
+  feat_strides:
+    - 8
+    - 16
+    - 32
+  hidden_dim: 256
+  num_levels: 3
+  num_layers: 6
+  num_queries: 300
+  num_denoising: 100
+  label_noise_ratio: 0.5
+  box_noise_scale: 1.0
+  eval_spatial_size:
+    - 640
+    - 640
+  eval_idx: -1
+  num_points:
+    - 4
+    - 4
+    - 4
+  cross_attn_method: default
+  query_select_method: default
+category:
+  - paragraph
+  - figure
+  - table

yomitoku-0.4.0.post1.dev0/configs/table_structure_recognitizer.yaml

@@ -0,0 +1,2 @@
+hf_hub_repo: "KotaroKinoshita/yomitoku-table-structure-recognizer-rtdtrv2-open-beta"
+thresh_score: 0.6

yomitoku-0.4.0.post1.dev0/configs/text_detector.yaml

@@ -0,0 +1,30 @@
+hf_hub_repo: KotaroKinoshita/yomitoku-text-detector-dbnet-open-beta
+backbone:
+  name: resnet50
+  dilation: true
+decoder:
+  in_channels:
+    - 256
+    - 512
+    - 1024
+    - 2048
+  hidden_dim: 256
+  adaptive: true
+  serial: true
+  smooth: false
+  k: 50
+data:
+  shortest_size: 1280
+  limit_size: 1600
+post_process:
+  min_size: 2
+  thresh: 0.2
+  box_thresh: 0.5
+  max_candidates: 1500
+  unclip_ratio: 2.0
+visualize:
+  color:
+    - 0
+    - 255
+    - 0
+  heatmap: false

yomitoku-0.4.0.post1.dev0/configs/text_recognizer.yaml

@@ -0,0 +1,14 @@
+hf_hub_repo: KotaroKinoshita/yomitoku-text-recognizer-parseq-open-beta
+max_label_length: 100
+decode_ar: 1
+refine_iters: 1
+data:
+  num_workers: 4
+  batch_size: 128
+visualize:
+  font: resource/MPLUS1p-Medium.ttf
+  color:
+    - 0
+    - 0
+    - 255
+  font_size: 18

yomitoku-0.4.0.post1.dev0/demo/setting_document_anaysis.py

@@ -0,0 +1,25 @@
+import cv2
+
+from yomitoku import DocumentAnalyzer
+from yomitoku.data.functions import load_pdf
+
+if __name__ == "__main__":
+    PATH_IMGE = ""
+
+    configs = {
+        "ocr": {"text_detector": {"path_cfg": "demo/text_detector.yaml"}}
+    }
+
+    analyzer = DocumentAnalyzer(configs=configs, visualize=True, device="cuda")
+
+    # PDFファイルを読み込み
+    imgs = load_pdf("demo/sample.pdf")
+    for i, img in enumerate(imgs):
+        results, ocr_vis, layout_vis = analyzer(img)
+
+        # HTML形式で解析結果をエクスポート
+        results.to_html(f"output_{i}.html")
+
+        # 可視化画像を保存
+        cv2.imwrite(f"output_ocr_{i}.jpg", ocr_vis)
+        cv2.imwrite(f"output_layout_{i}.jpg", layout_vis)

yomitoku-0.4.0.post1.dev0/demo/simple_document_analysis.py

@@ -0,0 +1,20 @@
+import cv2
+
+from yomitoku import DocumentAnalyzer
+from yomitoku.data.functions import load_pdf
+
+if __name__ == "__main__":
+    PATH_IMGE = ""
+
+    analyzer = DocumentAnalyzer(configs=None, visualize=True, device="cuda")
+    # PDFファイルを読み込み
+    imgs = load_pdf("demo/sample.pdf")
+    for i, img in enumerate(imgs):
+        results, ocr_vis, layout_vis = analyzer(img)
+
+        # HTML形式で解析結果をエクスポート
+        results.to_html(f"output_{i}.html")
+
+        # 可視化画像を保存
+        cv2.imwrite(f"output_ocr_{i}.jpg", ocr_vis)
+        cv2.imwrite(f"output_layout_{i}.jpg", layout_vis)

yomitoku-0.4.0.post1.dev0/demo/simple_layout.py

@@ -0,0 +1,15 @@
+import cv2
+
+from yomitoku import LayoutAnalyzer
+from yomitoku.data.functions import load_pdf
+
+if __name__ == "__main__":
+    analyzer = LayoutAnalyzer(configs=None, visualize=True, device="cuda")
+    # PDFファイルを読み込み
+    imgs = load_pdf("demo/sample.pdf")
+    for i, img in enumerate(imgs):
+        results, layout_vis = analyzer(img)
+
+        # JSON形式で解析結果をエクスポート
+        results.to_json(f"output_{i}.json")
+        cv2.imwrite(f"output_layout_{i}.jpg", layout_vis)

yomitoku-0.4.0.post1.dev0/demo/simple_ocr.py

@@ -0,0 +1,15 @@
+import cv2
+
+from yomitoku import OCR
+from yomitoku.data.functions import load_pdf
+
+if __name__ == "__main__":
+    ocr = OCR(configs=None, visualize=True, device="cuda")
+    # PDFファイルを読み込み
+    imgs = load_pdf("demo/sample.pdf")
+    for i, img in enumerate(imgs):
+        results, ocr_vis = ocr(img)
+
+        # JSON形式で解析結果をエクスポート
+        results.to_json(f"output_{i}.json")
+        cv2.imwrite(f"output_ocr_{i}.jpg", ocr_vis)

yomitoku-0.4.0.post1.dev0/demo/text_detector.yaml

@@ -0,0 +1,4 @@
+#hf_hub_repo: yomitoku-text-detector-dbnet-open-beta
+post_process:
+  thresh: 0.1
+  unclip_ratio: 2.5