1515您可以使用命令行将图片的类别分出来,命令行使用方式如下:
1616
1717``` bash
18- paddlex --pipeline image_classification --model PP-LCNet_x1_0 --input https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_image_classification_001.jpg
18+ paddlex --pipeline image_classification --model PP-LCNet_x1_0 --input https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_image_classification_001.jpg --device gpu:0
1919```
2020参数解释:
2121- ` pipeline ` : 产线名称,当前支持的产线名称有 ` image_classification ` 、` object_detection ` 、` semantic_segmentation ` 、` instance_segmentation ` 、` OCR ` 。
2222- ` model ` : 模型名称,每个产线支持的模型名称不同,请参考 [ PaddleX 模型库] ( ../models/support_model_list.md ) 。对于多模型组合的产线,需要指定多个模型名称,以空格分隔。
2323- ` input ` : 输入图片路径或 URL。
24-
24+ - ` device ` :训练设备,可选 ` cpu ` 、 ` gpu ` 、 ` xpu ` 、 ` npu ` 、 ` mlu ` ,除 cpu 外,可指定卡号,如: ` gpu:0 ` 。
2525
2626** Python API 使用方式**
2727
2828
2929``` python
30+ from pathlib import Path
3031from paddlex import ClsPipeline
3132from paddlex import PaddleInferenceOption
3233
34+ # 实例化 PaddleInferenceOption 设置推理配置
35+ kernel_option = PaddleInferenceOption()
36+ kernel_option.set_device(" gpu:0" )
37+
3338model_name = " PP-LCNet_x1_0"
34- pipeline = ClsPipeline(model_name, kernel_option = PaddleInferenceOption())
39+ output_base = Path(" output" )
40+ output = output_base / model_name
41+
42+ pipeline = ClsPipeline(model_name, output = output, kernel_option = kernel_option)
3543result = pipeline.predict(
3644 {' input_path' : " https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_image_classification_001.jpg" }
3745 )
@@ -49,13 +57,14 @@ print(result["cls_result"])
4957您可以使用命令行将图片中的目标检测出来,命令行使用方式如下:
5058
5159``` bash
52- paddlex --pipeline object_detection --model RT-DETR-L --input https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_object_detection_002.png
60+ paddlex --pipeline object_detection --model RT-DETR-L --input https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_object_detection_002.png --device gpu:0
5361
5462```
5563参数解释:
5664- ` pipeline ` : 产线名称,当前支持的产线名称有 ` image_classification ` 、` object_detection ` 、` semantic_segmentation ` 、` instance_segmentation ` 、` OCR ` 。
5765- ` model ` : 模型名称,每个产线支持的模型名称不同,请参考 [ PaddleX 模型库] ( ../models/support_model_list.md ) 。对于多模型组合的产线,需要指定多个模型名称,以空格分隔。
5866- ` input ` : 输入图片路径或 URL。
67+ - ` device ` :训练设备,可选` cpu ` 、` gpu ` 、` xpu ` 、` npu ` 、` mlu ` ,除 cpu 外,可指定卡号,如:` gpu:0 ` 。
5968
6069** Python API 使用方式**
6170
@@ -64,15 +73,18 @@ from pathlib import Path
6473from paddlex import DetPipeline
6574from paddlex import PaddleInferenceOption
6675
76+ # 实例化 PaddleInferenceOption 设置推理配置
77+ kernel_option = PaddleInferenceOption()
78+ kernel_option.set_device(" gpu:0" )
79+
6780model_name = " RT-DETR-L"
6881output_base = Path(" output" )
69-
7082output = output_base / model_name
71- pipeline = DetPipeline(model_name, output = output, kernel_option = PaddleInferenceOption())
83+
84+ pipeline = DetPipeline(model_name, output = output, kernel_option = kernel_option)
7285result = pipeline.predict(
7386 {" input_path" : " https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_object_detection_002.png" })
7487print (result[" boxes" ])
75-
7688```
7789
7890
@@ -86,13 +98,14 @@ print(result["boxes"])
8698您可以使用命令行将图片的语义信息分割出来,命令行使用方式如下:
8799
88100``` bash
89- paddlex --pipeline semantic_segmentation --model OCRNet_HRNet-W48 --input https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_semantic_segmentation_002.png
101+ paddlex --pipeline semantic_segmentation --model OCRNet_HRNet-W48 --input https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_semantic_segmentation_002.png --device gpu:0
90102
91103```
92104参数解释:
93105- ` pipeline ` : 产线名称,当前支持的产线名称有 ` image_classification ` 、` object_detection ` 、` semantic_segmentation ` 、` instance_segmentation ` 、` OCR ` 。
94106- ` model ` : 模型名称,每个产线支持的模型名称不同,请参考 [ PaddleX 模型库] ( ../models/support_model_list.md ) 。对于多模型组合的产线,需要指定多个模型名称,以空格分隔。
95107- ` input ` : 输入图片路径或 URL。
108+ - ` device ` :训练设备,可选` cpu ` 、` gpu ` 、` xpu ` 、` npu ` 、` mlu ` ,除 cpu 外,可指定卡号,如:` gpu:0 ` 。
96109
97110** Python API 使用方式**
98111
@@ -101,16 +114,19 @@ from pathlib import Path
101114from paddlex import SegPipeline
102115from paddlex import PaddleInferenceOption
103116
117+ # 实例化 PaddleInferenceOption 设置推理配置
118+ kernel_option = PaddleInferenceOption()
119+ kernel_option.set_device(" gpu:0" )
104120
105- model_name = " OCRNet_HRNet-W48" ,
121+ model_name = " OCRNet_HRNet-W48"
106122output_base = Path(" output" )
107123output = output_base / model_name
108- pipeline = SegPipeline(model_name, output = output, kernel_option = PaddleInferenceOption())
124+
125+ pipeline = SegPipeline(model_name, output = output, kernel_option = kernel_option)
109126result = pipeline.predict(
110127 {" input_path" : " https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_semantic_segmentation_002.png" }
111128)
112129print (result[" seg_map" ])
113-
114130```
115131
116132### 2.4 通用实例分割产线
@@ -123,13 +139,14 @@ print(result["seg_map"])
123139您可以使用命令行将图片中的实例分割出来,命令行使用方式如下:
124140
125141``` bash
126- paddlex --pipeline instance_segmentation --model Mask-RT-DETR-L --input https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_instance_segmentation_004.png
142+ paddlex --pipeline instance_segmentation --model Mask-RT-DETR-L --input https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_instance_segmentation_004.png --device gpu:0
127143
128144```
129145参数解释:
130146- ` pipeline ` : 产线名称,当前支持的产线名称有 ` image_classification ` 、` object_detection ` 、` semantic_segmentation ` 、` instance_segmentation ` 、` OCR ` 。
131147- ` model ` : 模型名称,每个产线支持的模型名称不同,请参考 [ PaddleX 模型库] ( ../models/support_model_list.md ) 。对于多模型组合的产线,需要指定多个模型名称,以空格分隔。
132148- ` input ` : 输入图片路径或 URL。
149+ - ` device ` :训练设备,可选` cpu ` 、` gpu ` 、` xpu ` 、` npu ` 、` mlu ` ,除 cpu 外,可指定卡号,如:` gpu:0 ` 。
133150
134151** Python API 使用方式**
135152
@@ -138,15 +155,18 @@ from pathlib import Path
138155from paddlex import InstanceSegPipeline
139156from paddlex import PaddleInferenceOption
140157
158+ # 实例化 PaddleInferenceOption 设置推理配置
159+ kernel_option = PaddleInferenceOption()
160+ kernel_option.set_device(" gpu:0" )
161+
141162model_name = " Mask-RT-DETR-L"
142163output_base = Path(" output" )
143-
144164output = output_base / model_name
145- pipeline = InstanceSegPipeline(model_name, output = output, kernel_option = PaddleInferenceOption())
165+
166+ pipeline = InstanceSegPipeline(model_name, output = output, kernel_option = kernel_option)
146167result = pipeline.predict(
147168 {" input_path" : " https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_instance_segmentation_004.png" })
148169print (result[" boxes" ])
149-
150170```
151171
152172### 2.5 OCR 产线
@@ -158,27 +178,32 @@ OCR 产线内置了 PP-OCRv4 模型,包括文本检测和文本识别两个部
158178您可以使用命令行将图片的文字识别出来,命令行使用方式如下:
159179
160180``` bash
161- paddlex --pipeline OCR --model PP-OCRv4_mobile_det PP-OCRv4_mobile_rec --input https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_ocr_002.png
181+ paddlex --pipeline OCR --model PP-OCRv4_mobile_det PP-OCRv4_mobile_rec --input https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_ocr_002.png --device gpu:0
162182```
163183参数解释:
164184- ` pipeline ` : 产线名称,当前支持的产线名称有 ` image_classification ` 、` object_detection ` 、` semantic_segmentation ` 、` instance_segmentation ` 、` OCR ` 。
165185- ` model ` : 模型名称,每个产线支持的模型名称不同,请参考 [ PaddleX 模型库] ( ../models/support_model_list.md ) 。对于多模型组合的产线,需要指定多个模型名称,以空格分隔。
166186- ` input ` : 输入图片路径或 URL。
167- </details >
187+ - ` device ` :训练设备,可选` cpu ` 、` gpu ` 、` xpu ` 、` npu ` 、` mlu ` ,除 cpu 外,可指定卡号,如:` gpu:0 ` 。
188+
168189
169190** Python API 使用方式**
170191
171192``` python
172193import cv2
173194from paddlex import OCRPipeline
174195from paddlex import PaddleInferenceOption
175- from paddlex.pipelines.PPOCR .utils import draw_ocr_box_txt
196+ from paddlex.pipelines.OCR .utils import draw_ocr_box_txt
197+
198+ # 实例化 PaddleInferenceOption 设置推理配置
199+ kernel_option = PaddleInferenceOption()
200+ kernel_option.set_device(" gpu:0" )
176201
177202pipeline = OCRPipeline(
178203 ' PP-OCRv4_mobile_det' ,
179204 ' PP-OCRv4_mobile_rec' ,
180- text_det_kernel_option = PaddleInferenceOption() ,
181- text_rec_kernel_option = PaddleInferenceOption() ,)
205+ text_det_kernel_option = kernel_option ,
206+ text_rec_kernel_option = kernel_option ,)
182207result = pipeline.predict(
183208 {" input_path" : " https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_ocr_002.png" },
184209)
0 commit comments