Skip to content

Commit 5720776

Browse files
authored
Update README.md
1 parent a0d0865 commit 5720776

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

README.md

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -215,15 +215,22 @@ you can use darknet2pytorch to convert it yourself, or download my converted mod
215215
# 7. ONNX2TensorRT and DeepStream Inference
216216
217217
1. Compile the DeepStream Nvinfer Plugin
218-
```cd DeepStream
218+
219+
```
220+
cd DeepStream
219221
make
220222
```
221223
2. Build a TRT Engine.
224+
222225
For single batch,
223-
```trtexec --onnx=<onnx_file> --explicitBatch --saveEngine=<tensorRT_engine_file> --workspace=<size_in_megabytes> --fp16```
226+
```
227+
trtexec --onnx=<onnx_file> --explicitBatch --saveEngine=<tensorRT_engine_file> --workspace=<size_in_megabytes> --fp16
228+
```
224229
225230
For multi-batch,
226-
```trtexec --onnx=<onnx_file> --explicitBatch --shapes=input:Xx3xHxW --optShapes=input:Xx3xHxW --maxShapes=input:Xx3xHxW --minShape=input:1x3xHxW --saveEngine=<tensorRT_engine_file> --fp16```
231+
```
232+
trtexec --onnx=<onnx_file> --explicitBatch --shapes=input:Xx3xHxW --optShapes=input:Xx3xHxW --maxShapes=input:Xx3xHxW --minShape=input:1x3xHxW --saveEngine=<tensorRT_engine_file> --fp16
233+
```
227234
228235
Note :The maxShapes could not be larger than model original shape.
229236

0 commit comments

Comments
 (0)