Skip to content

Commit a7193ca

Browse files
committed
docs: update files
1 parent 87a0f3e commit a7193ca

1 file changed

Lines changed: 17 additions & 17 deletions

File tree

docs/install_usage/how_to_use_other_engine.md

Lines changed: 17 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -41,43 +41,43 @@ results.vis("layout_res.png")
4141

4242
多卡时可通过 `cuda_ep_cfg.device_id` 指定卡号(与 [engine_cfg.yaml](https://github.com/RapidAI/RapidLayout/blob/main/rapid_layout/configs/engine_cfg.yaml)`cuda_ep_cfg.device_id` 一致)。
4343

44-
## 使用 OpenVINO
44+
## 使用 NPU (CANN)
4545

46-
```bash
47-
pip install rapid-layout onnxruntime openvino
48-
```
46+
详细配置参数参见:[engine_cfg.yaml](https://github.com/RapidAI/RapidLayout/blob/main/rapid_layout/configs/engine_cfg.yaml)
4947

5048
```python linenums="1"
51-
from rapid_layout import EngineType, ModelType, RapidLayout
49+
from rapid_layout import EngineType, ModelType, RapidLayout, RapidLayoutInput
5250

53-
layout_engine = RapidLayout(
51+
cfg = RapidLayoutInput(
5452
model_type=ModelType.PP_LAYOUT_CDLA,
55-
engine_type=EngineType.OPENVINO,
53+
engine_type=EngineType.ONNXRUNTIME,
54+
engine_cfg={"use_cann": True, "cann_ep_cfg": {"device_id": 0}},
5655
)
56+
layout_engine = RapidLayout(cfg=cfg)
57+
5758
img_path = "https://raw.githubusercontent.com/RapidAI/RapidLayout/718b60e927ab893c2fad67c98f753b2105a6f421/tests/test_files/layout.jpg"
5859
results = layout_engine(img_path)
5960
print(results)
6061
results.vis("layout_res.png")
6162
```
6263

63-
OpenVINO 设备与线程等配置见 [engine_cfg.yaml](https://github.com/RapidAI/RapidLayout/blob/main/rapid_layout/configs/engine_cfg.yaml)`openvino` 段。
64-
65-
## 使用 NPU (CANN)
64+
## 使用 OpenVINO
6665

67-
详细配置参数参见:[engine_cfg.yaml](https://github.com/RapidAI/RapidLayout/blob/main/rapid_layout/configs/engine_cfg.yaml)
66+
```bash
67+
pip install rapid-layout onnxruntime openvino
68+
```
6869

6970
```python linenums="1"
70-
from rapid_layout import EngineType, ModelType, RapidLayout, RapidLayoutInput
71+
from rapid_layout import EngineType, ModelType, RapidLayout
7172

72-
cfg = RapidLayoutInput(
73+
layout_engine = RapidLayout(
7374
model_type=ModelType.PP_LAYOUT_CDLA,
74-
engine_type=EngineType.ONNXRUNTIME,
75-
engine_cfg={"use_cann": True, "cann_ep_cfg": {"device_id": 0}},
75+
engine_type=EngineType.OPENVINO,
7676
)
77-
layout_engine = RapidLayout(cfg=cfg)
78-
7977
img_path = "https://raw.githubusercontent.com/RapidAI/RapidLayout/718b60e927ab893c2fad67c98f753b2105a6f421/tests/test_files/layout.jpg"
8078
results = layout_engine(img_path)
8179
print(results)
8280
results.vis("layout_res.png")
8381
```
82+
83+
OpenVINO 设备与线程等配置见 [engine_cfg.yaml](https://github.com/RapidAI/RapidLayout/blob/main/rapid_layout/configs/engine_cfg.yaml)`openvino` 段。

0 commit comments

Comments
 (0)