From 72ae571da36aff55058c70f302e57097ad6278fc Mon Sep 17 00:00:00 2001
From: Wenyu <wenyu.lyu@gmail.com>
Date: Wed, 19 Apr 2023 19:40:35 +0800
Subject: [PATCH] update readme (#8108)

---
 configs/rtdetr/README.md | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/configs/rtdetr/README.md b/configs/rtdetr/README.md
index e1f2a5be8..e68533e1a 100644
--- a/configs/rtdetr/README.md
+++ b/configs/rtdetr/README.md
@@ -32,7 +32,7 @@ RT-DETR是第一个实时端到端目标检测器。具体而言，我们设计
 <details open>
 <summary>依赖包:</summary>
 
-- PaddlePaddle == 2.4.1
+- PaddlePaddle >= 2.4.1
 
 </details>
 
@@ -85,7 +85,7 @@ python tools/infer.py -c configs/rtdetr/rtdetr_r50vd_6x_coco.yml \
 ### 导出及转换模型
 
 <details open>
-<summary>1. 导出模型</summary>
+<summary>1. 导出模型 </summary>
 
 ```shell
 cd PaddleDetection
@@ -97,7 +97,7 @@ python tools/export_model.py -c configs/rtdetr/rtdetr_r50vd_6x_coco.yml \
 </details>
 
 <details>
-<summary>2. 转换模型至ONNX (点击展开)</summary>
+<summary>2. 转换模型至ONNX </summary>
 
 - 安装[Paddle2ONNX](https://github.com/PaddlePaddle/Paddle2ONNX) 和 ONNX
 
@@ -116,6 +116,18 @@ paddle2onnx --model_dir=./output_inference/rtdetr_r50vd_6x_coco/ \
             --save_file rtdetr_r50vd_6x_coco.onnx
 ```
 
+- 转换成TensorRT（可选）:
+
+```shell
+# 保证TensorRT的版本>=8.5.1
+trtexec --onnx=./rtdetr_r50vd_6x_coco.onnx \
+        --workspace=4096 \
+        --shapes=image:1x3x640x640 \
+        --saveEngine=rtdetr_r50vd_6x_coco.trt \
+        --avgRuns=100 \
+        --fp16
+```
+
 </details>
 
 ## 引用RT-DETR
-- 
GitLab