mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2026-04-23 00:17:25 +08:00
yeliang2258
a509dd8ec1
* add smoke model * add 3d vis * update code * update doc * mv paddle3d from detection to perception * update result for velocity * update code for CI * add set input data for TRT backend * add serving support for smoke model * update code * update code * update code --------- Co-authored-by: DefTruth <31974251+DefTruth@users.noreply.github.com>
2.5 KiB
Executable File
2.5 KiB
Executable File
English | 简体中文
Smoke Python Deployment Example
Before deployment, the following two steps need to be confirmed
-
- The hardware and software environment meets the requirements, refer to FastDeploy environment requirements
-
- FastDeploy Python whl package installation, refer to FastDeploy Python Installation
This directory provides an example of infer.py to quickly complete the deployment of Smoke on CPU/GPU. Execute the following script to complete
#Download deployment sample code
git clone https://github.com/PaddlePaddle/FastDeploy.git
cd examples/vision/vision/paddle3d/smoke/python
wget https://bj.bcebos.com/fastdeploy/models/smoke.tar.gz
tar -xf smoke.tar.gz
wget https://bj.bcebos.com/fastdeploy/models/smoke_test.png
# CPU reasoning
python infer.py --model smoke --image smoke_test.png --device cpu
# GPU inference
python infer.py --model smoke --image smoke_test.png --device gpu
The visual result after running is shown in the figure below
Smoke Python interface
fastdeploy.vision.detection.Smoke(model_file, params_file, config_file, runtime_option=None, model_format=ModelFormat.PADDLE)
Smoke model loading and initialization.
parameter
- model_file(str): model file path
- params_file(str): parameter file path
- config_file(str): configuration file path
- runtime_option(RuntimeOption): Backend reasoning configuration, the default is None, that is, the default configuration is used
- model_format(ModelFormat): model format, the default is Paddle format
predict function
Smoke. predict(image_data)Model prediction interface, the input image directly outputs the detection result.
parameters
- image_data(np.ndarray): input data, note that it must be in HWC, BGR format
Back
Return the
fastdeploy.vision.PerceptionResultstructure, structure description reference document Vision Model Prediction Results