Skip to content

Latest commit

 

History

History
13 lines (7 loc) · 884 Bytes

ReadMe_CN.md

File metadata and controls

13 lines (7 loc) · 884 Bytes

Demo for using Forward-Cpp in Linux

  1. 参考 CMake-Build,构建 Forward-Cpp 库。

  2. 拷贝 build/bin 目录下的 libfwd_torch.so (libfwd_tf.solibfwd_keras.solibfwd_onnx.so) 和 libtrt_engine.solibs 目录下。

  3. 修改 build.sh 中的 TensorRT_ROOT。如果需要测试 test_fwd_torch.cpp 或者 test_fwd_tf.cpp 或者 test_fwd_onnx.cpp,则需要同时修改 ENABLE_TORCH=ONENABLE_TENSORFLOW=ONENABLE_ONNX=ON,以及 LibTorchLibTensorflow

  4. 更新 test_fwd_engine.cpp, test_fwd_tf.cpptest_fwd_torch.cpptest_fwd_onnx.cpp 中的 model_path and input informations

  5. 执行 build.sh 编译 demo 项目。

注意: 模型输入名可通过模型查看器来查看, 例如用 Netron 查看。