mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2026-04-23 00:17:25 +08:00
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.8 KiB
Executable File
2.8 KiB
Executable File
English | 简体中文
Smoke C++部署示例
本目录下提供 infer.cc 快速完成 Smoke 在 CPU/GPU 上部署的示例。
在部署前,需确认以下两个步骤
-
- 软硬件环境满足要求,参考FastDeploy环境要求
-
- 根据开发环境,下载预编译部署库和samples代码,参考FastDeploy预编译库
以Linux上 CPU 推理为例,在本目录执行如下命令即可完成编译测试,支持此模型需保证 FastDeploy 版本1.0.6以上(x.x.x>=1.0.6)
mkdir build
cd build
# 下载FastDeploy预编译库,用户可在上文提到的`FastDeploy预编译库`中自行选择合适的版本使用
wget https://bj.bcebos.com/fastdeploy/release/cpp/fastdeploy-linux-x64-x.x.x.tgz
tar xvf fastdeploy-linux-x64-x.x.x.tgz
cmake .. -DFASTDEPLOY_INSTALL_DIR=${PWD}/fastdeploy-linux-x64-x.x.x
make -j
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推理
./infer_demo smoke smoke_test.png 0
# GPU推理
./infer_demo smoke smoke_test.png 1
运行完成可视化结果如下图所示
以上命令只适用于Linux或MacOS, Windows下SDK的使用方式请参考:
Smoke C++ 接口
Smoke 类
fastdeploy::vision::detection::Smoke(
const string& model_file,
const string& params_file,
const string& config_file,
const RuntimeOption& runtime_option = RuntimeOption(),
const ModelFormat& model_format = ModelFormat::PADDLE)
Smoke模型加载和初始化。
参数
- model_file(str): 模型文件路径
- params_file(str): 参数文件路径
- config_file(str): 配置文件路径
- runtime_option(RuntimeOption): 后端推理配置,默认为None,即采用默认配置
- model_format(ModelFormat): 模型格式,默认为Paddle格式
Predict函数
Smoke::Predict(cv::Mat* im, PerceptionResult* result)模型预测接口,输入图像直接输出检测结果。
参数
- im: 输入图像,注意需为HWC,BGR格式
- result: 检测结果,包括检测框,各个框的置信度, PerceptionResult 说明参考视觉模型预测结果