749749 </ span >
750750 </ a >
751751
752+ </ li >
753+
754+ < li class ="md-nav__item ">
755+ < a href ="#npu " class ="md-nav__link ">
756+ < span class ="md-ellipsis ">
757+
758+ 使用NPU
759+
760+ </ span >
761+ </ a >
762+
752763</ li >
753764
754765 </ ul >
920931 </ span >
921932 </ a >
922933
934+ </ li >
935+
936+ < li class ="md-nav__item ">
937+ < a href ="#npu " class ="md-nav__link ">
938+ < span class ="md-ellipsis ">
939+
940+ 使用NPU
941+
942+ </ span >
943+ </ a >
944+
923945</ li >
924946
925947 </ ul >
@@ -1003,38 +1025,64 @@ <h2 id="gpu">使用GPU<a class="headerlink" href="#gpu" title="Permanent link">&
10031025</ span > < span id ="__span-0-5 "> < a id ="__codelineno-0-5 " name ="__codelineno-0-5 " href ="#__codelineno-0-5 "> </ a > pip< span class ="w "> </ span > install< span class ="w "> </ span > onnxruntime-gpu
10041026</ span > </ code > </ pre > </ div >
10051027< h2 id ="_2 "> 使用< a class ="headerlink " href ="#_2 " title ="Permanent link "> ¶</ a > </ h2 >
1006- < p > ```python linenum="1"
1007- from rapid_layout import EngineType, ModelType, RapidLayout, RapidLayoutInput</ p >
1008- < p > cfg = RapidLayoutInput(
1009- model_type=ModelType.PP_LAYOUT_CDLA,
1010- engine_type=EngineType.ONNXRUNTIME,
1011- engine_cfg={"use_cuda": True, "cuda_ep_cfg.gpu_id": 1},
1012- )
1013- layout_engine = RapidLayout(cfg=cfg)</ p >
1014- < p > img_path = "https://raw.githubusercontent.com/RapidAI/RapidLayout/718b60e927ab893c2fad67c98f753b2105a6f421/tests/test_files/layout.jpg"
1015- results = layout_engine(img_path)
1016- print(results)</ p >
1017- < p > results.vis("layout_res.png")
1018- < div class ="language-text highlight "> < pre > < span > </ span > < code > < span id ="__span-1-1 "> < a id ="__codelineno-1-1 " name ="__codelineno-1-1 " href ="#__codelineno-1-1 "> </ a > ## 使用NPU
1019- </ span > < span id ="__span-1-2 "> < a id ="__codelineno-1-2 " name ="__codelineno-1-2 " href ="#__codelineno-1-2 "> </ a >
1020- </ span > < span id ="__span-1-3 "> < a id ="__codelineno-1-3 " name ="__codelineno-1-3 " href ="#__codelineno-1-3 "> </ a > 详细配置参数参见:[engine_cfg.yaml](https://github.com/RapidAI/RapidLayout/blob/a7ab63ff291bd72e1a98ac2bb11860575514f432/rapid_layout/configs/engine_cfg.yaml)
1021- </ span > < span id ="__span-1-4 "> < a id ="__codelineno-1-4 " name ="__codelineno-1-4 " href ="#__codelineno-1-4 "> </ a >
1022- </ span > < span id ="__span-1-5 "> < a id ="__codelineno-1-5 " name ="__codelineno-1-5 " href ="#__codelineno-1-5 "> </ a > ```python linenum="1"
1023- </ span > < span id ="__span-1-6 "> < a id ="__codelineno-1-6 " name ="__codelineno-1-6 " href ="#__codelineno-1-6 "> </ a > from rapid_layout import EngineType, ModelType, RapidLayout, RapidLayoutInput
1024- </ span > < span id ="__span-1-7 "> < a id ="__codelineno-1-7 " name ="__codelineno-1-7 " href ="#__codelineno-1-7 "> </ a >
1025- </ span > < span id ="__span-1-8 "> < a id ="__codelineno-1-8 " name ="__codelineno-1-8 " href ="#__codelineno-1-8 "> </ a > cfg = RapidLayoutInput(
1026- </ span > < span id ="__span-1-9 "> < a id ="__codelineno-1-9 " name ="__codelineno-1-9 " href ="#__codelineno-1-9 "> </ a > model_type=ModelType.PP_LAYOUT_CDLA,
1027- </ span > < span id ="__span-1-10 "> < a id ="__codelineno-1-10 " name ="__codelineno-1-10 " href ="#__codelineno-1-10 "> </ a > engine_type=EngineType.ONNXRUNTIME,
1028- </ span > < span id ="__span-1-11 "> < a id ="__codelineno-1-11 " name ="__codelineno-1-11 " href ="#__codelineno-1-11 "> </ a > engine_cfg={"use_cann": True, "cann_ep_cfg.gpu_id": 0},
1029- </ span > < span id ="__span-1-12 "> < a id ="__codelineno-1-12 " name ="__codelineno-1-12 " href ="#__codelineno-1-12 "> </ a > )
1030- </ span > < span id ="__span-1-13 "> < a id ="__codelineno-1-13 " name ="__codelineno-1-13 " href ="#__codelineno-1-13 "> </ a > layout_engine = RapidLayout(cfg=cfg)
1031- </ span > < span id ="__span-1-14 "> < a id ="__codelineno-1-14 " name ="__codelineno-1-14 " href ="#__codelineno-1-14 "> </ a >
1032- </ span > < span id ="__span-1-15 "> < a id ="__codelineno-1-15 " name ="__codelineno-1-15 " href ="#__codelineno-1-15 "> </ a > img_path = "https://raw.githubusercontent.com/RapidAI/RapidLayout/718b60e927ab893c2fad67c98f753b2105a6f421/tests/test_files/layout.jpg"
1033- </ span > < span id ="__span-1-16 "> < a id ="__codelineno-1-16 " name ="__codelineno-1-16 " href ="#__codelineno-1-16 "> </ a > results = layout_engine(img_path)
1034- </ span > < span id ="__span-1-17 "> < a id ="__codelineno-1-17 " name ="__codelineno-1-17 " href ="#__codelineno-1-17 "> </ a > print(results)
1035- </ span > < span id ="__span-1-18 "> < a id ="__codelineno-1-18 " name ="__codelineno-1-18 " href ="#__codelineno-1-18 "> </ a >
1036- </ span > < span id ="__span-1-19 "> < a id ="__codelineno-1-19 " name ="__codelineno-1-19 " href ="#__codelineno-1-19 "> </ a > results.vis("layout_res.png")
1037- </ span > </ code > </ pre > </ div > </ p >
1028+ < div class ="language-python highlight "> < table class ="highlighttable "> < tr > < td class ="linenos "> < div class ="linenodiv "> < pre > < span > </ span > < span class ="normal "> < a href ="#__codelineno-1-1 "> 1</ a > </ span >
1029+ < span class ="normal "> < a href ="#__codelineno-1-2 "> 2</ a > </ span >
1030+ < span class ="normal "> < a href ="#__codelineno-1-3 "> 3</ a > </ span >
1031+ < span class ="normal "> < a href ="#__codelineno-1-4 "> 4</ a > </ span >
1032+ < span class ="normal "> < a href ="#__codelineno-1-5 "> 5</ a > </ span >
1033+ < span class ="normal "> < a href ="#__codelineno-1-6 "> 6</ a > </ span >
1034+ < span class ="normal "> < a href ="#__codelineno-1-7 "> 7</ a > </ span >
1035+ < span class ="normal "> < a href ="#__codelineno-1-8 "> 8</ a > </ span >
1036+ < span class ="normal "> < a href ="#__codelineno-1-9 "> 9</ a > </ span >
1037+ < span class ="normal "> < a href ="#__codelineno-1-10 "> 10</ a > </ span >
1038+ < span class ="normal "> < a href ="#__codelineno-1-11 "> 11</ a > </ span >
1039+ < span class ="normal "> < a href ="#__codelineno-1-12 "> 12</ a > </ span >
1040+ < span class ="normal "> < a href ="#__codelineno-1-13 "> 13</ a > </ span >
1041+ < span class ="normal "> < a href ="#__codelineno-1-14 "> 14</ a > </ span > </ pre > </ div > </ td > < td class ="code "> < div > < pre > < span > </ span > < code > < span id ="__span-1-1 "> < a id ="__codelineno-1-1 " name ="__codelineno-1-1 "> </ a > < span class ="kn "> from</ span > < span class ="w "> </ span > < span class ="nn "> rapid_layout</ span > < span class ="w "> </ span > < span class ="kn "> import</ span > < span class ="n "> EngineType</ span > < span class ="p "> ,</ span > < span class ="n "> ModelType</ span > < span class ="p "> ,</ span > < span class ="n "> RapidLayout</ span > < span class ="p "> ,</ span > < span class ="n "> RapidLayoutInput</ span >
1042+ </ span > < span id ="__span-1-2 "> < a id ="__codelineno-1-2 " name ="__codelineno-1-2 "> </ a >
1043+ </ span > < span id ="__span-1-3 "> < a id ="__codelineno-1-3 " name ="__codelineno-1-3 "> </ a > < span class ="n "> cfg</ span > < span class ="o "> =</ span > < span class ="n "> RapidLayoutInput</ span > < span class ="p "> (</ span >
1044+ </ span > < span id ="__span-1-4 "> < a id ="__codelineno-1-4 " name ="__codelineno-1-4 "> </ a > < span class ="n "> model_type</ span > < span class ="o "> =</ span > < span class ="n "> ModelType</ span > < span class ="o "> .</ span > < span class ="n "> PP_LAYOUT_CDLA</ span > < span class ="p "> ,</ span >
1045+ </ span > < span id ="__span-1-5 "> < a id ="__codelineno-1-5 " name ="__codelineno-1-5 "> </ a > < span class ="n "> engine_type</ span > < span class ="o "> =</ span > < span class ="n "> EngineType</ span > < span class ="o "> .</ span > < span class ="n "> ONNXRUNTIME</ span > < span class ="p "> ,</ span >
1046+ </ span > < span id ="__span-1-6 "> < a id ="__codelineno-1-6 " name ="__codelineno-1-6 "> </ a > < span class ="n "> engine_cfg</ span > < span class ="o "> =</ span > < span class ="p "> {</ span > < span class ="s2 "> "use_cuda"</ span > < span class ="p "> :</ span > < span class ="kc "> True</ span > < span class ="p "> ,</ span > < span class ="s2 "> "cuda_ep_cfg.gpu_id"</ span > < span class ="p "> :</ span > < span class ="mi "> 1</ span > < span class ="p "> },</ span >
1047+ </ span > < span id ="__span-1-7 "> < a id ="__codelineno-1-7 " name ="__codelineno-1-7 "> </ a > < span class ="p "> )</ span >
1048+ </ span > < span id ="__span-1-8 "> < a id ="__codelineno-1-8 " name ="__codelineno-1-8 "> </ a > < span class ="n "> layout_engine</ span > < span class ="o "> =</ span > < span class ="n "> RapidLayout</ span > < span class ="p "> (</ span > < span class ="n "> cfg</ span > < span class ="o "> =</ span > < span class ="n "> cfg</ span > < span class ="p "> )</ span >
1049+ </ span > < span id ="__span-1-9 "> < a id ="__codelineno-1-9 " name ="__codelineno-1-9 "> </ a >
1050+ </ span > < span id ="__span-1-10 "> < a id ="__codelineno-1-10 " name ="__codelineno-1-10 "> </ a > < span class ="n "> img_path</ span > < span class ="o "> =</ span > < span class ="s2 "> "https://raw.githubusercontent.com/RapidAI/RapidLayout/718b60e927ab893c2fad67c98f753b2105a6f421/tests/test_files/layout.jpg"</ span >
1051+ </ span > < span id ="__span-1-11 "> < a id ="__codelineno-1-11 " name ="__codelineno-1-11 "> </ a > < span class ="n "> results</ span > < span class ="o "> =</ span > < span class ="n "> layout_engine</ span > < span class ="p "> (</ span > < span class ="n "> img_path</ span > < span class ="p "> )</ span >
1052+ </ span > < span id ="__span-1-12 "> < a id ="__codelineno-1-12 " name ="__codelineno-1-12 "> </ a > < span class ="nb "> print</ span > < span class ="p "> (</ span > < span class ="n "> results</ span > < span class ="p "> )</ span >
1053+ </ span > < span id ="__span-1-13 "> < a id ="__codelineno-1-13 " name ="__codelineno-1-13 "> </ a >
1054+ </ span > < span id ="__span-1-14 "> < a id ="__codelineno-1-14 " name ="__codelineno-1-14 "> </ a > < span class ="n "> results</ span > < span class ="o "> .</ span > < span class ="n "> vis</ span > < span class ="p "> (</ span > < span class ="s2 "> "layout_res.png"</ span > < span class ="p "> )</ span >
1055+ </ span > </ code > </ pre > </ div > </ td > </ tr > </ table > </ div >
1056+ < h2 id ="npu "> 使用NPU< a class ="headerlink " href ="#npu " title ="Permanent link "> ¶</ a > </ h2 >
1057+ < p > 详细配置参数参见:< a href ="https://github.com/RapidAI/RapidLayout/blob/a7ab63ff291bd72e1a98ac2bb11860575514f432/rapid_layout/configs/engine_cfg.yaml "> engine_cfg.yaml</ a > </ p >
1058+ < div class ="language-python highlight "> < table class ="highlighttable "> < tr > < td class ="linenos "> < div class ="linenodiv "> < pre > < span > </ span > < span class ="normal "> < a href ="#__codelineno-2-1 "> 1</ a > </ span >
1059+ < span class ="normal "> < a href ="#__codelineno-2-2 "> 2</ a > </ span >
1060+ < span class ="normal "> < a href ="#__codelineno-2-3 "> 3</ a > </ span >
1061+ < span class ="normal "> < a href ="#__codelineno-2-4 "> 4</ a > </ span >
1062+ < span class ="normal "> < a href ="#__codelineno-2-5 "> 5</ a > </ span >
1063+ < span class ="normal "> < a href ="#__codelineno-2-6 "> 6</ a > </ span >
1064+ < span class ="normal "> < a href ="#__codelineno-2-7 "> 7</ a > </ span >
1065+ < span class ="normal "> < a href ="#__codelineno-2-8 "> 8</ a > </ span >
1066+ < span class ="normal "> < a href ="#__codelineno-2-9 "> 9</ a > </ span >
1067+ < span class ="normal "> < a href ="#__codelineno-2-10 "> 10</ a > </ span >
1068+ < span class ="normal "> < a href ="#__codelineno-2-11 "> 11</ a > </ span >
1069+ < span class ="normal "> < a href ="#__codelineno-2-12 "> 12</ a > </ span >
1070+ < span class ="normal "> < a href ="#__codelineno-2-13 "> 13</ a > </ span >
1071+ < span class ="normal "> < a href ="#__codelineno-2-14 "> 14</ a > </ span > </ pre > </ div > </ td > < td class ="code "> < div > < pre > < span > </ span > < code > < span id ="__span-2-1 "> < a id ="__codelineno-2-1 " name ="__codelineno-2-1 "> </ a > < span class ="kn "> from</ span > < span class ="w "> </ span > < span class ="nn "> rapid_layout</ span > < span class ="w "> </ span > < span class ="kn "> import</ span > < span class ="n "> EngineType</ span > < span class ="p "> ,</ span > < span class ="n "> ModelType</ span > < span class ="p "> ,</ span > < span class ="n "> RapidLayout</ span > < span class ="p "> ,</ span > < span class ="n "> RapidLayoutInput</ span >
1072+ </ span > < span id ="__span-2-2 "> < a id ="__codelineno-2-2 " name ="__codelineno-2-2 "> </ a >
1073+ </ span > < span id ="__span-2-3 "> < a id ="__codelineno-2-3 " name ="__codelineno-2-3 "> </ a > < span class ="n "> cfg</ span > < span class ="o "> =</ span > < span class ="n "> RapidLayoutInput</ span > < span class ="p "> (</ span >
1074+ </ span > < span id ="__span-2-4 "> < a id ="__codelineno-2-4 " name ="__codelineno-2-4 "> </ a > < span class ="n "> model_type</ span > < span class ="o "> =</ span > < span class ="n "> ModelType</ span > < span class ="o "> .</ span > < span class ="n "> PP_LAYOUT_CDLA</ span > < span class ="p "> ,</ span >
1075+ </ span > < span id ="__span-2-5 "> < a id ="__codelineno-2-5 " name ="__codelineno-2-5 "> </ a > < span class ="n "> engine_type</ span > < span class ="o "> =</ span > < span class ="n "> EngineType</ span > < span class ="o "> .</ span > < span class ="n "> ONNXRUNTIME</ span > < span class ="p "> ,</ span >
1076+ </ span > < span id ="__span-2-6 "> < a id ="__codelineno-2-6 " name ="__codelineno-2-6 "> </ a > < span class ="n "> engine_cfg</ span > < span class ="o "> =</ span > < span class ="p "> {</ span > < span class ="s2 "> "use_cann"</ span > < span class ="p "> :</ span > < span class ="kc "> True</ span > < span class ="p "> ,</ span > < span class ="s2 "> "cann_ep_cfg.gpu_id"</ span > < span class ="p "> :</ span > < span class ="mi "> 0</ span > < span class ="p "> },</ span >
1077+ </ span > < span id ="__span-2-7 "> < a id ="__codelineno-2-7 " name ="__codelineno-2-7 "> </ a > < span class ="p "> )</ span >
1078+ </ span > < span id ="__span-2-8 "> < a id ="__codelineno-2-8 " name ="__codelineno-2-8 "> </ a > < span class ="n "> layout_engine</ span > < span class ="o "> =</ span > < span class ="n "> RapidLayout</ span > < span class ="p "> (</ span > < span class ="n "> cfg</ span > < span class ="o "> =</ span > < span class ="n "> cfg</ span > < span class ="p "> )</ span >
1079+ </ span > < span id ="__span-2-9 "> < a id ="__codelineno-2-9 " name ="__codelineno-2-9 "> </ a >
1080+ </ span > < span id ="__span-2-10 "> < a id ="__codelineno-2-10 " name ="__codelineno-2-10 "> </ a > < span class ="n "> img_path</ span > < span class ="o "> =</ span > < span class ="s2 "> "https://raw.githubusercontent.com/RapidAI/RapidLayout/718b60e927ab893c2fad67c98f753b2105a6f421/tests/test_files/layout.jpg"</ span >
1081+ </ span > < span id ="__span-2-11 "> < a id ="__codelineno-2-11 " name ="__codelineno-2-11 "> </ a > < span class ="n "> results</ span > < span class ="o "> =</ span > < span class ="n "> layout_engine</ span > < span class ="p "> (</ span > < span class ="n "> img_path</ span > < span class ="p "> )</ span >
1082+ </ span > < span id ="__span-2-12 "> < a id ="__codelineno-2-12 " name ="__codelineno-2-12 "> </ a > < span class ="nb "> print</ span > < span class ="p "> (</ span > < span class ="n "> results</ span > < span class ="p "> )</ span >
1083+ </ span > < span id ="__span-2-13 "> < a id ="__codelineno-2-13 " name ="__codelineno-2-13 "> </ a >
1084+ </ span > < span id ="__span-2-14 "> < a id ="__codelineno-2-14 " name ="__codelineno-2-14 "> </ a > < span class ="n "> results</ span > < span class ="o "> .</ span > < span class ="n "> vis</ span > < span class ="p "> (</ span > < span class ="s2 "> "layout_res.png"</ span > < span class ="p "> )</ span >
1085+ </ span > </ code > </ pre > </ div > </ td > </ tr > </ table > </ div >
10381086
10391087<!-- Source file information -->
10401088
@@ -1059,7 +1107,7 @@ <h2 id="_2">使用<a class="headerlink" href="#_2" title="Permanent link">¶
10591107 < span class ="md-icon " title ="最后更新 ">
10601108 < svg xmlns ="http://www.w3.org/2000/svg " viewBox ="0 0 24 24 "> < path d ="M21 13.1c-.1 0-.3.1-.4.2l-1 1 2.1 2.1 1-1c.2-.2.2-.6 0-.8l-1.3-1.3c-.1-.1-.2-.2-.4-.2m-1.9 1.8-6.1 6V23h2.1l6.1-6.1zM12.5 7v5.2l4 2.4-1 1L11 13V7zM11 21.9c-5.1-.5-9-4.8-9-9.9C2 6.5 6.5 2 12 2c5.3 0 9.6 4.1 10 9.3-.3-.1-.6-.2-1-.2s-.7.1-1 .2C19.6 7.2 16.2 4 12 4c-4.4 0-8 3.6-8 8 0 4.1 3.1 7.5 7.1 7.9l-.1.2z "/> </ svg >
10611109 </ span >
1062- < span class ="git-revision-date-localized-plugin git-revision-date-localized-plugin-iso_datetime " title ="2026年2月7日 11:11:53 CST "> 2026-02-07 11:11:53 </ span >
1110+ < span class ="git-revision-date-localized-plugin git-revision-date-localized-plugin-iso_datetime " title ="2026年2月7日 11:19:40 CST "> 2026-02-07 11:19:40 </ span >
10631111 </ span >
10641112
10651113
0 commit comments