mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-12-24 13:28:13 +08:00
hjyp
cc4bbf2163
* add java examples * fix detail * fix pre-config --------- Co-authored-by: DefTruth <31974251+DefTruth@users.noreply.github.com>
56 lines
2.1 KiB
Markdown
56 lines
2.1 KiB
Markdown
[English](README.md) | 简体中文
|
|
# YOLOv5 Java 部署示例
|
|
|
|
本目录下提供`java/InferDemo.java`, 使用`Java`调用`C++`API快速完成`PaddleDetection`模型`YOLOv5`在Linux上部署的示例。
|
|
|
|
在部署前，需确认以下两个步骤
|
|
|
|
- 1. 软硬件环境满足要求，参考[FastDeploy环境要求](../../../../../docs/cn/build_and_install/download_prebuilt_libraries.md)
|
|
- 2. 根据开发环境，下载预编译部署库和samples代码，参考[FastDeploy预编译库](../../../../../docs/cn/build_and_install/download_prebuilt_libraries.md)
|
|
|
|
|
|
|
|
使用`Java`调用`C++` API 可以分为两步：
|
|
|
|
* 在`C++`端生成动态链接库。
|
|
* 在`Java`端调用动态链接库。
|
|
|
|
## C++端生成动态链接库
|
|
首先，切换路径到`cpp`目录，将`jdk`目录下的`jni.h`和`jni_md.h`拷贝到当前`cpp`目录下。
|
|
```shell
|
|
cp /PathJdk/jdk-17.0.6/include/jni.h ./
|
|
cp /Pathjdk/jdk-17.0.6/include/linux/jni_md.h ./
|
|
```
|
|
|
|
接着，在`cpp`目录下执行以下命令，进行编译，生成动态链接库。
|
|
> 注意：编译时需要通过`FASTDEPLOY_INSTALL_DIR`选项指明`FastDeploy`预编译库位置, 当然也可以是自己编译的`FastDeploy`库位置。
|
|
```shell
|
|
mkdir build && cd build
|
|
cmake .. -FASTDEPLOY_INSTALL_DIR /fast-deploy-path
|
|
make -j
|
|
```
|
|
编译成功后，动态链接库会存放在`cpp/build`目录下，`Linux`下以`.so`结尾，`Windows`下以`.dll`结尾。
|
|
|
|
## 使用JAVA调用动态链接库
|
|
|
|
将`FastDeploy`的库路径添加到环境变量，注意替换为自己的`FastDeploy`库所在路径。
|
|
```bash
|
|
source /Path/to/fastdeploy-linux-x64-1.0.4/fastdeploy_init.sh
|
|
```
|
|
下载YOLOv5模型文件和测试图片
|
|
```bash
|
|
wget https://bj.bcebos.com/paddlehub/fastdeploy/yolov5s.onnx
|
|
wget https://gitee.com/paddlepaddle/PaddleDetection/raw/release/2.4/demo/000000014439.jpg
|
|
```
|
|
|
|
编译Java文件
|
|
```shell
|
|
javac InferDemo.java
|
|
```
|
|
|
|
编译完成后，执行如下命令可得到预测结果，其中第一个参数指明下载的模型路径，第二个参数指明测试图片路径。
|
|
```shell
|
|
java InferDemo ./yolov5s.onnx ./000000014439.jpg
|
|
```
|
|
可视化的检测结果图片保存在本地`vis_result.jpg`。
|