This tutorial will introduce the detailed steps of deploying the PaddleClas classification model on the server side. The deployment of the recognition model will be supported in the near future. Please look forward to it.
- Linux, docker is recommended.
- Windows, compilation based on
Visual Studio 2019 Community
is supported. In addition, you can refer to How to use PaddleDetection to make a complete project to compile by generating thesln solution
. - This document mainly introduces the compilation and inference of PaddleClas using C++ in Linux environment.
- First of all, you need to download the source code compiled package in the Linux environment from the opencv official website. Taking opencv3.4.7 as an example, the download and uncompress command are as follows.
wget https://github.com/opencv/opencv/archive/3.4.7.tar.gz
tar -xf 3.4.7.tar.gz
Finally, you can see the folder of opencv-3.4.7/
in the current directory.
- Compile opencv, the opencv source path (
root_path
) and installation path (install_path
) should be set by yourself. Among them,root_path
is the downloaded opencv source code path, andinstall_path
is the installation path of opencv. In this case, the opencv source is./opencv-3.4.7
.
cd ./opencv-3.4.7
export root_path=$PWD
export install_path=${root_path}/opencv3
- After entering the opencv source code path, you can compile it in the following way.
rm -rf build
mkdir build
cd build
cmake .. \
-DCMAKE_INSTALL_PREFIX=${install_path} \
-DCMAKE_BUILD_TYPE=Release \
-DBUILD_SHARED_LIBS=OFF \
-DWITH_IPP=OFF \
-DBUILD_IPP_IW=OFF \
-DWITH_LAPACK=OFF \
-DWITH_EIGEN=OFF \
-DCMAKE_INSTALL_LIBDIR=lib64 \
-DWITH_ZLIB=ON \
-DBUILD_ZLIB=ON \
-DWITH_JPEG=ON \
-DBUILD_JPEG=ON \
-DWITH_PNG=ON \
-DBUILD_PNG=ON \
-DWITH_TIFF=ON \
-DBUILD_TIFF=ON
make -j
make install
- After
make install
is completed, the opencv header file and library file will be generated in this folder for later PaddleClas source code compilation.
Take opencv3.4.7 for example, the final file structure under the opencv installation path is as follows. NOTICE:The following file structure may be different for different Versions of Opencv.
opencv3/
|-- bin
|-- include
|-- lib64
|-- share
- There are 2 ways to obtain the Paddle Inference Library, described in detail below.
- If you want to get the latest Paddle Inference Library features, you can download the latest code from Paddle GitHub repository and compile the inference library from the source code.
- You can refer to Paddle Inference Library to get the Paddle source code from github, and then compile To generate the latest inference library. The method of using git to access the code is as follows.
git clone https://github.com/PaddlePaddle/Paddle.git
- After entering the Paddle directory, the compilation method is as follows.
rm -rf build
mkdir build
cd build
cmake .. \
-DWITH_CONTRIB=OFF \
-DWITH_MKL=ON \
-DWITH_MKLDNN=ON \
-DWITH_TESTING=OFF \
-DCMAKE_BUILD_TYPE=Release \
-DWITH_INFERENCE_API_TEST=OFF \
-DON_INFER=ON \
-DWITH_PYTHON=ON
make -j
make inference_lib_dist
For more compilation parameter options, please refer to the official website of the Paddle C++ inference library:https://www.paddlepaddle.org.cn/documentation/docs/en/develop/guides/05_inference_deployment/inference/build_and_install_lib_en.html#build-from-source-code.
- After the compilation process, you can see the following files in the folder of
build/paddle_inference_install_dir/
.
build/paddle_inference_install_dir/
|-- CMakeCache.txt
|-- paddle
|-- third_party
|-- version.txt
Among them, paddle
is the Paddle library required for C++ prediction later, and version.txt
contains the version information of the current inference library.
-
Different cuda versions of the Linux inference library (based on GCC 4.8.2) are provided on the Paddle Inference Library official website. You can view and select the appropriate version of the inference library on the official website.
-
Please select the
develop
version. -
After downloading, use the following method to uncompress.
tar -xf paddle_inference.tgz
Finally you can see the following files in the folder of paddle_inference/
.
- The compilation commands are as follows. The addresses of Paddle C++ inference library, opencv and other Dependencies need to be replaced with the actual addresses on your own machines.
sh tools/build.sh
Specifically, the content in tools/build.sh
is as follows.
OPENCV_DIR=your_opencv_dir
LIB_DIR=your_paddle_inference_dir
CUDA_LIB_DIR=your_cuda_lib_dir
CUDNN_LIB_DIR=your_cudnn_lib_dir
TENSORRT_DIR=your_tensorrt_lib_dir
BUILD_DIR=build
rm -rf ${BUILD_DIR}
mkdir ${BUILD_DIR}
cd ${BUILD_DIR}
cmake .. \
-DPADDLE_LIB=${LIB_DIR} \
-DWITH_MKL=ON \
-DDEMO_NAME=clas_system \
-DWITH_GPU=OFF \
-DWITH_STATIC_LIB=OFF \
-DWITH_TENSORRT=OFF \
-DTENSORRT_DIR=${TENSORRT_DIR} \
-DOPENCV_DIR=${OPENCV_DIR} \
-DCUDNN_LIB=${CUDNN_LIB_DIR} \
-DCUDA_LIB=${CUDA_LIB_DIR} \
make -j
In the above parameters of command:
-
OPENCV_DIR
is the opencv installation path; -
LIB_DIR
is the download (paddle_inference
folder) or the generated Paddle Inference Library path (build/paddle_inference_install_dir
folder); -
CUDA_LIB_DIR
is the cuda library file path, in docker; it is/usr/local/cuda/lib64
; -
CUDNN_LIB_DIR
is the cudnn library file path, in docker it is/usr/lib/x86_64-linux-gnu/
. -
TENSORRT_DIR
is the tensorrt library file path,in dokcer it is/usr/local/TensorRT6-cuda10.0-cudnn7/
,TensorRT is just enabled for GPU.
After the compilation is completed, an executable file named clas_system
will be generated in the build
folder.
In addition to compiling the demo directly, you can also compile only config lib and cls lib by running the following command:
sh tools/build_lib.sh
The contents of the above command are as follows:
OpenCV_DIR=path/to/opencv
PADDLE_LIB_DIR=path/to/paddle
BUILD_DIR=./lib/build
rm -rf ${BUILD_DIR}
mkdir ${BUILD_DIR}
cd ${BUILD_DIR}
cmake .. \
-DOpenCV_DIR=${OpenCV_DIR} \
-DPADDLE_LIB=${PADDLE_LIB_DIR} \
-DCMP_STATIC=ON \
make
The specific description of each compilation option is as follows:
DOpenCV_DIR
: The directory to the OpenCV compilation library. In this example, it isopencv-3.4.7/opencv3/share/OpenCV
. Note that there needs to be aOpenCVConfig.cmake
file under this directory;DPADDLE_LIB
: The directory to the paddle prediction library which generally is the path ofpaddle_inference
downloaded and decompressed or compiled, such asbuild/paddle_inference_install_dir
. Note that there should be two subdirectoriespaddle
andthird_party
in this directory;DCMP_STATIC
: Whether to compile config lib and cls lib into static link library (.a
). The default isON
. If you need to compile into dynamic link library (.so
), please set it toOFF
.
After executing the above commands, the dynamic link libraries (libcls.so
and libconfig.so
) or static link libraries (cls.a
and libconfig.a
) of config lib and cls lib will be generated in the directory. In the 2.1 Compile PaddleClas C++ inference demo, you can specify the compilation option DCLS_LIB
and DCONFIG_LIB
to the path of the existing link library of cls lib
and config lib
, which can also be used for development.
- You can refer to Model inference,export the inference model. After the model is exported, assuming it is placed in the
inference
directory, the directory structure is as follows.
inference/
|--cls_infer.pdmodel
|--cls_infer.pdiparams
NOTICE: Among them, cls_infer.pdmodel
file stores the model structure information and the cls_infer.pdiparams
file stores the model parameter information.The paths of the two files need to correspond to the parameters of cls_model_path
and cls_params_path
in the configuration file tools/config.txt
.
First, please modify the tools/config.txt
and tools/run.sh
.
-
Some key words in
tools/config.txt
is as follows.- use_gpu: Whether to use GPU.
- gpu_id: GPU id.
- gpu_mem:GPU memory.
- cpu_math_library_num_threads:Number of thread for math library acceleration.
- use_mkldnn:Whether to use mkldnn.
- use_tensorrt: Whether to use tensorRT.
- use_fp16:Whether to use Float16 (half precision), it is just enabled when use_tensorrt is set as 1.
- cls_model_path: Model path of inference model.
- cls_params_path: Params path of inference model.
- resize_short_size:Short side length of the image after resize.
- crop_size:Image size after center crop.
-
You could modify
tools/run.sh
(./build/clas_system ./tools/config.txt ./docs/imgs/ILSVRC2012_val_00000666.JPEG
):- ./build/clas_system: the path of executable file compiled;
- ./tools/config.txt: the path of config;
- ./docs/imgs/ILSVRC2012_val_00000666.JPEG: the path of image file to be predicted.
-
Then execute the following command to complete the classification of an image.
sh tools/run.sh
- The prediction results will be shown on the screen, which is as follows.
- In the above results,
class id
represents the id corresponding to the category with the highest confidence, andscore
represents the probability that the image belongs to that category.