yomitoku 0.6.0__tar.gz → 0.7.1__tar.gz

{yomitoku-0.6.0 → yomitoku-0.7.1}/.github/workflows/lint-and-test.yml

@@ -30,6 +30,8 @@ jobs:
           enable-cache: true
       - name: pin python version
         run: uv python pin 3.12
+      - name: Update apt
+        run: sudo apt update
       - name: Install dependencies
         run: sudo apt install -y libopencv-dev poppler-utils
       - name: Install tox-uv
@@ -37,4 +39,4 @@ jobs:
       - name: Run linter
         run: tox -e lint
       - name: Run tests
-        run:  tox -p -e py310,py311,py312
+        run: tox -p -e py310,py311,py312

{yomitoku-0.6.0 → yomitoku-0.7.1}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: yomitoku
-Version: 0.6.0
+Version: 0.7.1
 Summary: Yomitoku is an AI-powered document image analysis package designed specifically for the Japanese language.
 Author-email: Kotaro Kinoshita <kotaro.kinoshita@mlism.com>
 License: CC BY-NC-SA 4.0
@@ -10,7 +10,6 @@ Requires-Dist: huggingface-hub>=0.26.1
 Requires-Dist: lxml>=5.3.0
 Requires-Dist: omegaconf>=2.3.0
 Requires-Dist: onnx>=1.17.0
-Requires-Dist: onnxruntime-gpu>=1.20.1
 Requires-Dist: onnxruntime>=1.20.1
 Requires-Dist: opencv-python>=4.10.0.84
 Requires-Dist: pyclipper>=1.3.0.post6
@@ -20,6 +19,8 @@ Requires-Dist: shapely>=2.0.6
 Requires-Dist: timm>=1.0.11
 Requires-Dist: torch>=2.5.0
 Requires-Dist: torchvision>=0.20.0
+Provides-Extra: gpu
+Requires-Dist: onnxruntime-gpu>=1.20.1; extra == 'gpu'
 Description-Content-Type: text/markdown
 
 日本語版 | [English](README_EN.md)
@@ -72,6 +73,11 @@ Markdown でエクスポートした結果は関してはリポジトリ内の[s
 pip install yomitoku
 ```
 
+onnxruntimeの実行にGPUを使用する場合
+```
+pip install yomitoku[gpu]
+```
+
 - pytorch はご自身の CUDA のバージョンにあったものをインストールしてください。デフォルトでは CUDA12.4 以上に対応したものがインストールされます。
 - pytorch は 2.5 以上のバージョンに対応しています。その関係で CUDA11.8 以上のバージョンが必要になります。対応できない場合は、リポジトリ内の Dockerfile を利用してください。
 
@@ -89,7 +95,8 @@ yomitoku ${path_data} -f md -o results -v --figure　--lite
 - `-d`, `--device` モデルを実行するためのデバイスを指定します。gpu が利用できない場合は cpu で推論が実行されます。(デフォルト: cuda)
 - `--ignore_line_break` 画像の改行位置を無視して、段落内の文章を連結して返します。（デフォルト：画像通りの改行位置位置で改行します。）
 - `--figure_letter` 検出した図表に含まれる文字も出力ファイルにエクスポートします。
-- `--figure` 検出した図、画像を出力ファイルにエクスポートします。(html と markdown のみ)
+- `--figure` 検出した図、画像を出力ファイルにエクスポートします。
+- `--encoding` エクスポートする出力ファイルの文字エンコーディングを指定します。サポートされていない文字コードが含まれる場合は、その文字を無視します。(utf-8, utf-8-sig, shift-jis, enc-jp, cp932)
 
 その他のオプションに関しては、ヘルプを参照
 

{yomitoku-0.6.0 → yomitoku-0.7.1}/README.md

@@ -48,6 +48,11 @@ Markdown でエクスポートした結果は関してはリポジトリ内の[s
 pip install yomitoku
 ```
 
+onnxruntimeの実行にGPUを使用する場合
+```
+pip install yomitoku[gpu]
+```
+
 - pytorch はご自身の CUDA のバージョンにあったものをインストールしてください。デフォルトでは CUDA12.4 以上に対応したものがインストールされます。
 - pytorch は 2.5 以上のバージョンに対応しています。その関係で CUDA11.8 以上のバージョンが必要になります。対応できない場合は、リポジトリ内の Dockerfile を利用してください。
 
@@ -65,7 +70,8 @@ yomitoku ${path_data} -f md -o results -v --figure　--lite
 - `-d`, `--device` モデルを実行するためのデバイスを指定します。gpu が利用できない場合は cpu で推論が実行されます。(デフォルト: cuda)
 - `--ignore_line_break` 画像の改行位置を無視して、段落内の文章を連結して返します。（デフォルト：画像通りの改行位置位置で改行します。）
 - `--figure_letter` 検出した図表に含まれる文字も出力ファイルにエクスポートします。
-- `--figure` 検出した図、画像を出力ファイルにエクスポートします。(html と markdown のみ)
+- `--figure` 検出した図、画像を出力ファイルにエクスポートします。
+- `--encoding` エクスポートする出力ファイルの文字エンコーディングを指定します。サポートされていない文字コードが含まれる場合は、その文字を無視します。(utf-8, utf-8-sig, shift-jis, enc-jp, cp932)
 
 その他のオプションに関しては、ヘルプを参照
 

{yomitoku-0.6.0 → yomitoku-0.7.1}/README_EN.md

@@ -49,6 +49,10 @@ Source of the image: Created by processing content from “Reiwa 6 Edition Infor
 pip install yomitoku
 ```
 
+Using GPU with onnxruntime
+```
+pip install yomitoku[gpu]
+```
 
 - Please install the version of PyTorch that matches your CUDA version. By default, a version compatible with CUDA 12.4 or higher will be installed.
 - PyTorch versions 2.5 and above are supported. As a result, CUDA version 11.8 or higher is required. If this is not feasible, please use the Dockerfile provided in the repository.
@@ -67,7 +71,8 @@ yomitoku ${path_data} -f md -o results -v --figure
 - `-d`, `--device`: Specify the device for running the model. If a GPU is unavailable, inference will be executed on the CPU. (Default: cuda)
 - `--ignore_line_break`: Ignores line breaks in the image and concatenates sentences within a paragraph. (Default: respects line breaks as they appear in the image.)
 - `--figure_letter`: Exports characters contained within detected figures and tables to the output file.
-- `--figure`: Exports detected figures and images to the output file (supported only for html and markdown).
+- `--figure`: Exports detected figures and images to the output file 
+- `--encoding` Specifies the character encoding for the output file to be exported. If unsupported characters are included, they will be ignored. (utf-8, utf-8-sig, shift-jis, enc-jp, cp932)
 
 
 For other options, please refer to the help documentation.

yomitoku-0.7.1/demo/setting_document_anaysis.py

@@ -0,0 +1,6 @@
+from yomitoku import DocumentAnalyzer
+
+if __name__ == "__main__":
+    configs = {"ocr": {"text_detector": {"path_cfg": "demo/text_detector.yaml"}}}
+
+    analyzer = DocumentAnalyzer(configs=configs, visualize=True, device="cuda")

{yomitoku-0.6.0 → yomitoku-0.7.1}/demo/simple_document_analysis.py

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

{yomitoku-0.6.0 → yomitoku-0.7.1}/demo/simple_layout.py

@@ -4,7 +4,7 @@ from yomitoku import LayoutAnalyzer
 from yomitoku.data.functions import load_pdf
 
 if __name__ == "__main__":
-    analyzer = LayoutAnalyzer(configs=None, visualize=True, device="cuda")
+    analyzer = LayoutAnalyzer(visualize=True, device="cuda")
     # PDFファイルを読み込み
     imgs = load_pdf("demo/sample.pdf")
     for i, img in enumerate(imgs):

{yomitoku-0.6.0 → yomitoku-0.7.1}/demo/simple_ocr.py

@@ -4,7 +4,7 @@ from yomitoku import OCR
 from yomitoku.data.functions import load_pdf
 
 if __name__ == "__main__":
-    ocr = OCR(configs=None, visualize=True, device="cuda")
+    ocr = OCR(visualize=True, device="cuda")
     # PDFファイルを読み込み
     imgs = load_pdf("demo/sample.pdf")
     for i, img in enumerate(imgs):

{yomitoku-0.6.0 → yomitoku-0.7.1}/docs/installation.en.md

@@ -9,6 +9,12 @@ This package requires Python 3.10 or later and PyTorch 2.5 or later for executio
 pip install yomitoku
 ```
 
+Using GPU with onnxruntime
+```bash
+pip install yomitoku[gpu]
+```
+
+
 ## using uv
 This repository uses the package management tool [uv](https://docs.astral.sh/uv/). After installing uv, clone the repository and execute the following commands:
 
@@ -16,6 +22,11 @@ This repository uses the package management tool [uv](https://docs.astral.sh/uv/
 uv sync
 ```
 
+Using GPU with onnxruntime
+```bash
+uv sync --extra gpu
+```
+
 When using uv, you need to modify the following part of the pyproject.toml file to match your CUDA version. By default, PyTorch compatible with CUDA 12.4 will be downloaded.
 
 ```pyproject.tom

{yomitoku-0.6.0 → yomitoku-0.7.1}/docs/installation.ja.md

@@ -8,6 +8,11 @@
 pip install yomitoku
 ```
 
+onnxruntimeの実行にGPUを使用する場合
+```bash
+pip install yomitoku[gpu]
+```
+
 ## uv でのインストール
 
 本リポジトリはパッケージ管理ツールに [uv](https://docs.astral.sh/uv/) を使用しています。uv をインストール後、リポジトリをクローンし、以下のコマンドを実行してください
@@ -16,6 +21,11 @@ pip install yomitoku
 uv sync
 ```
 
+onnxruntimeの実行にGPUを使用する場合
+```bash
+uv sync --extra gpu
+```
+
 uvを利用する場合、`pyproject.toml`の以下の部分をご自身のcudaのバージョンに合わせて修正する必要があります。デフォルトではCUDA12.4に対応したpytorchがダウンロードされます。
 
 ```pyproject.tom

{yomitoku-0.6.0 → yomitoku-0.7.1}/docs/usage.en.md

@@ -16,9 +16,12 @@ yomitoku ${path_data} -f md -o results -v
 - `-d`, `--device`: Specify the device for running the model. If a GPU is unavailable, inference will be executed on the CPU. (Default: cuda)
 - `--ignore_line_break`: Ignores line breaks in the image and concatenates sentences within a paragraph. (Default: respects line breaks as they appear in the image.)
 - `--figure_letter`: Exports characters contained within detected figures and tables to the output file.
-- `--figure`: Exports detected figures and images to the output file (supported only for html and markdown).
+- `--figure`: Exports detected figures and images to the output file
+- `--encoding` Specifies the character encoding for the output file to be exported. If unsupported characters are included, they will be ignored. (utf-8, utf-8-sig, shift-jis, enc-jp, cp932)
+
 
 **NOTE**
+
 - It is recommended to run on a GPU. The system is not optimized for inference on CPUs, which may result in significantly longer processing times.
 - Only printed text recognition is supported. While it may occasionally read handwritten text, official support is not provided.
 - YomiToku is optimized for document OCR and is not designed for scene OCR (e.g., text printed on non-paper surfaces like signs).
@@ -30,24 +33,11 @@ yomitoku ${path_data} -f md -o results -v
 
 The Document Analyzer performs OCR and layout analysis, integrating these results into a comprehensive analysis output. It can be used for various use cases, including paragraph and table structure analysis, extraction, and figure/table detection.
 
-```python
-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)
+<!--codeinclude-->
 
-    # HTML形式で解析結果をエクスポート
-    results.to_html(PATH_OUTPUT)
+[demo/simple_document_analysis.py](../demo/simple_document_analysis.py)
 
-    # 可視化画像を保存
-    cv2.imwrite("output_ocr.jpg", ocr_vis)
-    cv2.imwrite("output_layout.jpg", layout_vis)
-```
+<!--/codeinclude-->
 
 - Setting `visualize` to True enables the visualization of each processing result. The second and third return values will contain the OCR and layout analysis results, respectively. If set to False, None will be returned. Since visualization adds computational overhead, it is recommended to set it to False unless needed for debugging purposes.
 - The `device` parameter specifies the computation device to be used. The default is "cuda". If a GPU is unavailable, it automatically switches to CPU mode for processing.
@@ -55,31 +45,20 @@ if __name__ == "__main__":
 
 The results of DocumentAnalyzer can be exported in the following formats:
 
-`to_json()`: JSON format (*.json)
-`to_html()`: HTML format (*.html)
-`to_csv()`: Comma-separated CSV format (*.csv)
-`to_markdown()`: Markdown format (*.md)
-
+`to_json()`: JSON format (_.json)
+`to_html()`: HTML format (_.html)
+`to_csv()`: Comma-separated CSV format (_.csv)
+`to_markdown()`: Markdown format (_.md)
 
 ### Using AI-OCR Only
 
-AI-OCR performs text detection and recognition on the detected text, returning the positions of the text within the image along with the recognition results.
+AI-OCR performs text detection and recognition on the detected text, returning the positions of the text within the image along with the
 
-```python
-import cv2
+<!--codeinclude-->
 
-from yomitoku import OCR
-from yomitoku.data.functions import load_image
+[demo/simple_ocr.py](../demo/simple_ocr.py)
 
-if __name__ == "__main__":
-    img = load_image(PATH_IMAGE)
-    ocr = OCR(configs=None, visualize=True, device="cuda")
-    results, ocr_vis = ocr(img)
-
-    # JSON形式で解析結果をエクスポート
-    results.to_json(PATH_OUTPUT)
-    cv2.imwrite("output_ocr.jpg", ocr_vis)
-```
+<!--/codeinclude-->
 
 - Setting `visualize` to True enables the visualization of each processing result. The second and third return values will contain the OCR and layout analysis results, respectively. If set to False, None will be returned. Since visualization adds computational overhead, it is recommended to set it to False unless needed for debugging purposes.
 - The `device` parameter specifies the computation device to be used. The default is "cuda". If a GPU is unavailable, it automatically switches to CPU mode for processing.
@@ -91,22 +70,11 @@ The results of OCR processing support export in JSON format (`to_json()`) only.
 
 The `LayoutAnalyzer` performs text detection, followed by AI-based paragraph, figure/table detection, and table structure analysis. It analyzes the layout structure within the document.
 
-```python
-import cv2
+<!--codeinclude-->
 
-from yomitoku import LayoutAnalyzer
-from yomitoku.data.functions import load_image
-
-if __name__ == "__main__":
-    img = load_image(PATH_IMAGE)
-    analyzer = LayoutAnalyzer(configs=None, visualize=True, device="cuda")
-    results, layout_vis = analyzer(img)
-
-    # JSON形式で解析結果をエクスポート
-    results.to_json(PATH_OUTPUT)
-    cv2.imwrite("output_layout.jpg", layout_vis)
-```
+[demo/simple_layout.py](../demo/simple_layout.py)
 
+<!--/codeinclude-->
 
 - Setting `visualize` to True enables the visualization of each processing result. The second and third return values will contain the OCR and layout analysis results, respectively. If set to False, None will be returned. Since visualization adds computational overhead, it is recommended to set it to False unless needed for debugging purposes.
 - The `device` parameter specifies the computation device to be used. The default is "cuda". If a GPU is unavailable, it automatically switches to CPU mode for processing.
@@ -150,7 +118,6 @@ if __name__ == "__main__":
 
 ### Defining Parameters in an YAML File
 
-
 By providing the path to a YAML file in the config, you can adjust detailed parameters for inference. Examples of YAML files can be found in the `configs` directory within the repository. While the model's network parameters cannot be modified, certain aspects like post-processing parameters and input image size can be adjusted.
 
 For instance, you can define post-processing thresholds for the Text Detector in a YAML file and set its path in the config. The config file does not need to include all parameters; you only need to specify the parameters that require changes.
@@ -163,21 +130,11 @@ post_process:
 
 Storing the Path to a YAML File in the Config
 
-```python
-from yomitoku import DocumentAnalyzer
+<!--codeinclude-->
 
-if __name__ == "__main__":
-    # path_cfgに設定したymalのパスを記述する
-    configs = {
-        "ocr": {
-            "text_detector": {
-                "path_cfg": "text_detector.yaml"
-            }
-        }
-    }
+[demo/setting_document_anaysis.py](../demo/setting_document_anaysis.py)
 
-    DocumentAnalyzer(configs=configs)
-```
+<!--/codeinclude-->
 
 ## Using in an Offline Environment
 
@@ -186,7 +143,6 @@ Yomitoku automatically downloads models from Hugging Face Hub during the first e
 1. Install [Git Large File Storage](https://docs.github.com/ja/repositories/working-with-files/managing-large-files/installing-git-large-file-storage)
 2. In an environment with internet access, download the model repository. Copy the cloned repository to your target environment using your preferred tools.
 
-
 The following is the command to download the model repository from Hugging Face Hub.
 
 ```sh
@@ -207,18 +163,8 @@ hf_hub_repo: yomitoku-text-detector-dbnet-open-beta
 
 4. Storing the Path to a YAML File in the Config
 
-```python
-from yomitoku import DocumentAnalyzer
+<!--codeinclude-->
 
-if __name__ == "__main__":
-    # path_cfgに設定したymalのパスを記述する
-    configs = {
-        "ocr": {
-            "text_detector": {
-                "path_cfg": "text_detector.yaml"
-            }
-        }
-    }
+[demo/setting_document_anaysis.py](../demo/setting_document_anaysis.py)
 
-    DocumentAnalyzer(configs=configs)
-```
+<!--/codeinclude-->

{yomitoku-0.6.0 → yomitoku-0.7.1}/docs/usage.ja.md

@@ -8,12 +8,18 @@
 yomitoku ${path_data} -f md -o results -v
 ```
 
-- `${path_data}` 解析対象の画像が含まれたディレクトリか画像ファイルのパスを直接して指定してください。かディレクトリを対象とした場合はディレクトリのサブディレクトリ内の画像も含めて処理を実行します。入力をサポートしているファイル形式は jpeg, png, bmp, tiff, pdf です。
-- `-f` 出力形式のファイルフォーマットを指定します。(json, csv, html, md をサポート)
-- `-l` 指定すると軽量モデルで推論を実行します。CPUでも高速に推論可能です。
-- `-o` 出力先のディレクトリ名を指定します。存在しない場合は新規で作成されます。
-- `-v` を指定すると解析結果を可視化した画像を出力します。
-- `-d` モデルを実行するためのデバイスを指定します。gpu が利用できない場合は cpu で推論が実行されます。(デフォルト: cuda)
+- `${path_data}` 解析対象の画像が含まれたディレクトリか画像ファイルのパスを直接して指定してください。ディレクトリを対象とした場合はディレクトリのサブディレクトリ内の画像も含めて処理を実行します。
+- `-f`, `--format` 出力形式のファイルフォーマットを指定します。(json, csv, html, md をサポート)
+- `-o`, `--outdir` 出力先のディレクトリ名を指定します。存在しない場合は新規で作成されます。
+- `-v`, `--vis` を指定すると解析結果を可視化した画像を出力します。
+- `-l`, `--lite` を指定すると軽量モデルで推論を実行します。通常より高速に推論できますが、若干、精度が低下する可能性があります。
+- `-d`, `--device` モデルを実行するためのデバイスを指定します。gpu が利用できない場合は cpu で推論が実行されます。(デフォルト: cuda)
+- `--ignore_line_break` 画像の改行位置を無視して、段落内の文章を連結して返します。（デフォルト：画像通りの改行位置位置で改行します。）
+- `--figure_letter` 検出した図表に含まれる文字も出力ファイルにエクスポートします。
+- `--figure` 検出した図、画像を出力ファイルにエクスポートします。
+- `--encoding` エクスポートする出力ファイルの文字エンコーディングを指定します。サポートされていない文字コードが含まれる場合は、その文字を無視します。(utf-8, utf-8-sig, shift-jis, enc-jp, cp932)
+
+その他のオプションに関しては、ヘルプを参照
 
 ### Note:
 
@@ -28,24 +34,11 @@ yomitoku ${path_data} -f md -o results -v
 
 Document Analyzer は OCR およびレイアウト解析を実行し、それらの結果を統合した解析結果を返却します。段落、表の構造解析、抽出、図表の検知など様々なユースケースにご利用いただけます。
 
-```python
-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)
+<!--codeinclude-->
 
-    # HTML形式で解析結果をエクスポート
-    results.to_html(PATH_OUTPUT)
+[demo/simple_document_analysis.py](../demo/simple_document_analysis.py)
 
-    # 可視化画像を保存
-    cv2.imwrite("output_ocr.jpg", ocr_vis)
-    cv2.imwrite("output_layout.jpg", layout_vis)
-```
+<!--/codeinclude-->
 
 - `visualize` を True にすると各処理結果を可視化した結果を第２、第 3 戻り値に OCR、レアウト解析の処理結果をそれぞれ格納し、返却します。False にした場合は None を返却します。描画処理のための計算が増加しますので、デバック用途でない場合は、False を推奨します。
 - `device` には処理に用いる計算機を指定します。Default は"cuda". GPU が利用できない場合は、自動で CPU モードに切り替えて処理を実行します。
@@ -62,21 +55,11 @@ if __name__ == "__main__":
 
 AI-OCR では、テキスト検知と検知したテキストに対して、認識処理を実行し、画像内の文字の位置と読み取り結果を返却します。
 
-```python
-import cv2
-
-from yomitoku import OCR
-from yomitoku.data.functions import load_image
+<!--codeinclude-->
 
-if __name__ == "__main__":
-    img = load_image(PATH_IMAGE)
-    ocr = OCR(configs=None, visualize=True, device="cuda")
-    results, ocr_vis = ocr(img)
+[demo/simple_ocr.py](../demo/simple_ocr.py)
 
-    # JSON形式で解析結果をエクスポート
-    results.to_json(PATH_OUTPUT)
-    cv2.imwrite("output_ocr.jpg", ocr_vis)
-```
+<!--/codeinclude-->
 
 - `visualize` を True にすると各処理結果を可視化した結果を第２、第 3 戻り値に OCR、レアウト解析の処理結果をそれぞれ格納し、返却します。False にした場合は None を返却します。描画処理のための計算が増加しますので、デバック用途でない場合は、False を推奨します。
 - `device` には処理に用いる計算機を指定します。Default は"cuda". GPU が利用できない場合は、自動で CPU モードに切り替えて処理を実行します。
@@ -88,21 +71,11 @@ if __name__ == "__main__":
 
 LayoutAnalyzer では、テキスト検知と検知したテキストに対して、段落、図表の検知および表の構造解析処理 AI を実行し、文書内のレイアウト構造を解析します。
 
-```python
-import cv2
-
-from yomitoku import LayoutAnalyzer
-from yomitoku.data.functions import load_image
+<!--codeinclude-->
 
-if __name__ == "__main__":
-    img = load_image(PATH_IMAGE)
-    analyzer = LayoutAnalyzer(configs=None, visualize=True, device="cuda")
-    results, layout_vis = analyzer(img)
+[demo/simple_layout.py](../demo/simple_layout.py)
 
-    # JSON形式で解析結果をエクスポート
-    results.to_json(PATH_OUTPUT)
-    cv2.imwrite("output_layout.jpg", layout_vis)
-```
+<!--/codeinclude-->
 
 - `visualize` を True にすると各処理結果を可視化した結果を第２、第 3 戻り値に OCR、レアウト解析の処理結果をそれぞれ格納し、返却します。False にした場合は None を返却します。描画処理のための計算が増加しますので、デバック用途でない場合は、False を推奨します。
 - `device` には処理に用いる計算機を指定します。Default は"cuda". GPU が利用できない場合は、自動で CPU モードに切り替えて処理を実行します。
@@ -160,21 +133,11 @@ post_process:
 
 yaml ファイルのパスを config に格納する
 
-```python
-from yomitoku import DocumentAnalyzer
+<!--codeinclude-->
 
-if __name__ == "__main__":
-    # path_cfgに設定したymalのパスを記述する
-    configs = {
-        "ocr": {
-            "text_detector": {
-                "path_cfg": "text_detector.yaml"
-            }
-        }
-    }
+[demo/setting_document_anaysis.py](../demo/setting_document_anaysis.py)
 
-    DocumentAnalyzer(configs=configs)
-```
+<!--/codeinclude-->
 
 ## インターネットに接続できない環境での利用
 
@@ -204,18 +167,8 @@ hf_hub_repo: yomitoku-text-detector-dbnet-open-beta
 
 4. yaml ファイルのパスを config に格納する
 
-```python
-from yomitoku import DocumentAnalyzer
+<!--codeinclude-->
 
-if __name__ == "__main__":
-    # path_cfgに設定したymalのパスを記述する
-    configs = {
-        "ocr": {
-            "text_detector": {
-                "path_cfg": "text_detector.yaml"
-            }
-        }
-    }
+[demo/setting_document_anaysis.py](../demo/setting_document_anaysis.py)
 
-    DocumentAnalyzer(configs=configs)
-```
+<!--/codeinclude-->

{yomitoku-0.6.0 → yomitoku-0.7.1}/mkdocs.yml

@@ -73,6 +73,7 @@ repo_url: https://github.com/kotaro-kinoshita/yomitoku-dev
 
 plugins:
   - search:
+  - codeinclude:
   - i18n:
       languages:
         - locale: ja

{yomitoku-0.6.0 → yomitoku-0.7.1}/pyproject.toml

@@ -27,7 +27,13 @@ dependencies = [
     "torch>=2.5.0",
     "pypdfium2>=4.30.0",
     "onnx>=1.17.0",
+    "torch>=2.5.0",
+    "torchvision>=0.20.0",
     "onnxruntime>=1.20.1",
+]
+
+[project.optional-dependencies]
+gpu = [
     "onnxruntime-gpu>=1.20.1",
 ]
 
@@ -38,6 +44,7 @@ format = "{base}"
 
 [tool.uv]
 dev-dependencies = [
+    "mkdocs-codeinclude-plugin>=0.2.1",
     "mkdocs-material>=9.5.44",
     "mkdocs-static-i18n",
     "mkdocs>=1.6.1",
@@ -50,11 +57,9 @@ dev-dependencies = [
 
 [tool.uv.sources]
 torch = [
-    { index = "pytorch-cuda124", marker = "platform_system == 'Windows'"},
     { index = "pytorch-cpu", marker = "platform_system == 'Darwin'"},
 ]
 torchvision = [
-    { index = "pytorch-cuda124", marker = "platform_system == 'Windows'"},
     { index = "pytorch-cpu", marker = "platform_system == 'Darwin'"},
 ]
 
@@ -94,6 +99,7 @@ deps =
     pygments 
     plantuml_markdown 
     mkdocs-static-i18n
+    mkdocs-codeinclude-plugin
 commands = 
     mkdocs gh-deploy --force
 """