-
Build Forward-Cpp libraries refer to CMake-Build.
-
Copy
libfwd_torch.so
(libfwd_tf.so
,libfwd_keras.so
,libfwd_onnx.so
) andlibtrt_engine.so
from thebuild/bin
to thelibs
directory. -
Change the path of
TensorRT_ROOT
in the scriptbuild.sh
. If you need testtest_fwd_torch.cpp
or/andtest_fwd_tf.cpp
or/andtest_fwd_onnx.cpp
, then you should specifiedENABLE_TORCH=ON
or/andENABLE_TENSORFLOW=ON
or/andENABLE_ONNX=ON
, andLibTorch
or/andLibTensorflow
. -
Update the
model_path
andinput informations
in thetest_fwd_engine.cpp
,test_fwd_tf.cpp
or/andtest_fwd_torch.cpp
or/andtest_fwd_onnx.cpp
. -
run the script
build.sh
to build the demo project.
Notice: The name of INPUT in models can be viewed by model viewers, such as Netron.