@@ -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+
5758img_path = " https://raw.githubusercontent.com/RapidAI/RapidLayout/718b60e927ab893c2fad67c98f753b2105a6f421/tests/test_files/layout.jpg"
5859results = layout_engine(img_path)
5960print (results)
6061results.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-
7977img_path = " https://raw.githubusercontent.com/RapidAI/RapidLayout/718b60e927ab893c2fad67c98f753b2105a6f421/tests/test_files/layout.jpg"
8078results = layout_engine(img_path)
8179print (results)
8280results.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