From b2a6d0bf4304463bd88a0f3256ea04ca88654a1f Mon Sep 17 00:00:00 2001 From: huangyifei <1144358492@qq.com> Date: Sun, 20 Nov 2022 21:11:01 -0500 Subject: [PATCH] add missing steps or comments --- README.md | 1 + deploy/README.md | 6 ++++-- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index ffc1305..6adcc18 100644 --- a/README.md +++ b/README.md @@ -37,6 +37,7 @@ Clone this code to your workspace. We call this directory as `$CLRNET_ROOT` ```Shell git clone https://github.com/Turoad/clrnet +git submodule update --init --recursive ``` ### Create a conda virtual environment and activate it (conda is optional) diff --git a/deploy/README.md b/deploy/README.md index 76f3f2f..ef1f70d 100644 --- a/deploy/README.md +++ b/deploy/README.md @@ -22,13 +22,15 @@ optional arguments: ``` For example: -``` +```shell python deploy/onnx/export_onnx.py --cfg configs/clrnet/clr_resnet18_culane.py \ --load-from culane_r18.pth \ --opset-version 16 \ --out-file culane_r18.onnx ``` +if meet error, you can try to use lower opset-version, eg: when you use the provided trt7.2 version docker, you should set opset-version <= 12. + ## Export from ONNX to TensorRT ``` @@ -95,7 +97,7 @@ export ONNXRUNTIME_DIR=$(pwd)/onnxruntime-linux-x64-1.8.1 ### Build the operator for ONNX and TensorRT ``` cd /path of CLRNet/deploy/csrc/ -mkdir build +mkdir build && cd build cmake .. make ```