x86.md 3.9 KB
Newer Older
1
# PaddleLite使用X86预测部署
2

3
Paddle-Lite 支持在Docker或Linux环境编译x86预测库。环境搭建参考[环境准备](../user_guides/source_compile)
4 5 6 7 8

## 编译

1、 下载代码
```bash
9
# 下载Paddle-Lite源码
10
git clone https://github.com/PaddlePaddle/Paddle-Lite.git
11
# 切换到release分支
12
git checkout release/v2.6.0
13 14 15 16 17 18 19
```

2、 源码编译

```bash
cd Paddle-Lite
./lite/tools/build.sh x86
20 21 22

# 其他可选择编译选项
# --with_log=OFF 关闭LOG信息输出
23 24 25 26 27 28 29 30 31 32 33 34 35
```

## 编译结果说明

x86编译结果位于 `build.lite.x86/inference_lite_lib`
**具体内容**说明:

1、 `bin`文件夹:可执行工具文件 `test_model_bin`

2、 `cxx`文件夹:包含c++的库文件与相应的头文件

- `include`  : 头文件
- `lib` : 库文件
36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52
  - 静态库文件:
    - `libpaddle_api_full_bundled.a`  :full_api 静态库
    - `libpaddle_api_light_bundled.a` :light_api 静态库
  - 动态库文件:
    - `libpaddle_full_api_shared.so` :full_api 动态库
    - `libpaddle_light_api_shared.so`:light_api 动态库

3、 `third_party` 文件夹:依赖的第三方预测库mklml

- mklml : Paddle-Lite预测库依赖的mklml数学库

4、 `demo/cxx`文件夹:x86预测库的C++ 示例demo

- `mobilenetv1_full` :使用full_api 执行mobilenet_v1预测的C++ demo
- `mobilenetv1_light` :使用light_api 执行mobilenet_v1预测的C++ demo


53 54 55 56


## x86预测API使用示例

57
1、`mobilenetv1_full`目录结构
58

59 60 61 62 63 64
```bash
mobilenetv1_full/
|-- CMakeLists.txt
|-- build.sh
`-- mobilenet_full_api.cc
```
65

66
本demo使用cmake构建`CMakeLists.txt`为cmake脚本,`mobilenet_full_api.cc`是x86示例的源代码、`build.sh`为编译的脚本。
67

68
2、demo使用方法
69 70 71

``` bash
# 1、编译
72
cd mobilenetv1_full
73 74 75 76 77
sh build.sh
```
编译结果为当前目录下的 `mobilenet_full_api `
``` bash
# 2、执行预测
78
./mobilenet_full_api ./mobilenet_v1
79
```
80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97
下载并解压模型[`mobilenet_v1`](http://paddle-inference-dist.bj.bcebos.com/mobilenet_v1.tar.gz)到当前目录,执行以上命令进行预测。

```bash
# 3、执行demo后输出结果如下,全一输入下mobilenet_v1的预测结果
Output shape 1000
Output[0]: 0.000191312
Output[100]: 0.000159713
Output[200]: 0.000264313
Output[300]: 0.000210793
Output[400]: 0.00103236
Output[500]: 0.000110071
Output[600]: 0.00482924
Output[700]: 0.00184533
Output[800]: 0.000202116
Output[900]: 0.000585591
```


98 99 100

3、示例源码`mobilenet_full_api.cc`

101 102 103
```c++
#include <iostream>
#include <vector>
104
#include "paddle_api.h"
105 106


107
using namespace paddle::lite_api;  // NOLINT
108 109 110 111 112 113 114

int64_t ShapeProduction(const shape_t& shape) {
  int64_t res = 1;
  for (auto i : shape) res *= i;
  return res;
}

115 116 117 118 119 120 121 122
void RunModel(std::string model_dir) {
   // 1. Create CxxConfig
   CxxConfig config;
   config.set_model_dir(model_dir);
   config.set_valid_places({
     Place{TARGET(kX86), PRECISION(kFloat)},
     Place{TARGET(kHost), PRECISION(kFloat)}
   });
123 124 125 126 127 128
  // 2. Create PaddlePredictor by CxxConfig
  std::shared_ptr<PaddlePredictor> predictor =
      CreatePaddlePredictor<CxxConfig>(config);

  // 3. Prepare input data
  std::unique_ptr<Tensor> input_tensor(std::move(predictor->GetInput(0)));
129
  input_tensor->Resize({1, 3, 224, 224});
130 131 132 133 134 135 136 137 138 139 140
  auto* data = input_tensor->mutable_data<float>();
  for (int i = 0; i < ShapeProduction(input_tensor->shape()); ++i) {
    data[i] = 1;
  }

  // 4. Run predictor
  predictor->Run();

  // 5. Get output
  std::unique_ptr<const Tensor> output_tensor(
      std::move(predictor->GetOutput(0)));
141
  std::cout << "Output shape " << output_tensor->shape()[1] << std::endl;
142
  for (int i = 0; i < ShapeProduction(output_tensor->shape()); i += 100) {
143 144
    std::cout << "Output[" << i << "]: " << output_tensor->data<float>()[i]
              << std::endl;
145 146 147 148
  }
}

int main(int argc, char** argv) {
149 150 151 152 153 154
  if (argc < 2) {
    std::cerr << "[ERROR] usage: ./" << argv[0] << " naive_buffer_model_dir\n";
    exit(1);
  }
  std::string model_dir = argv[1];
  RunModel(model_dir);
155 156
  return 0;
}
157

158
```