Files
FastDeploy/examples/vision/perception/paddle3d/smoke/python
yeliang2258 a509dd8ec1 [Model] Add Paddle3D smoke model (#1766)
* 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>
2023-04-14 16:30:56 +08:00
..

English | 简体中文

Smoke Python Deployment Example

Before deployment, the following two steps need to be confirmed

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.PerceptionResult structure, structure description reference document Vision Model Prediction Results

Other documents