未验证 提交 72ae571d 编写于 作者: W Wenyu 提交者: GitHub

update readme (#8108)

上级 4406f68e
...@@ -32,7 +32,7 @@ RT-DETR是第一个实时端到端目标检测器。具体而言,我们设计 ...@@ -32,7 +32,7 @@ RT-DETR是第一个实时端到端目标检测器。具体而言,我们设计
<details open> <details open>
<summary>依赖包:</summary> <summary>依赖包:</summary>
- PaddlePaddle == 2.4.1 - PaddlePaddle >= 2.4.1
</details> </details>
...@@ -85,7 +85,7 @@ python tools/infer.py -c configs/rtdetr/rtdetr_r50vd_6x_coco.yml \ ...@@ -85,7 +85,7 @@ python tools/infer.py -c configs/rtdetr/rtdetr_r50vd_6x_coco.yml \
### 导出及转换模型 ### 导出及转换模型
<details open> <details open>
<summary>1. 导出模型</summary> <summary>1. 导出模型 </summary>
```shell ```shell
cd PaddleDetection cd PaddleDetection
...@@ -97,7 +97,7 @@ python tools/export_model.py -c configs/rtdetr/rtdetr_r50vd_6x_coco.yml \ ...@@ -97,7 +97,7 @@ python tools/export_model.py -c configs/rtdetr/rtdetr_r50vd_6x_coco.yml \
</details> </details>
<details> <details>
<summary>2. 转换模型至ONNX (点击展开)</summary> <summary>2. 转换模型至ONNX </summary>
- 安装[Paddle2ONNX](https://github.com/PaddlePaddle/Paddle2ONNX) 和 ONNX - 安装[Paddle2ONNX](https://github.com/PaddlePaddle/Paddle2ONNX) 和 ONNX
...@@ -116,6 +116,18 @@ paddle2onnx --model_dir=./output_inference/rtdetr_r50vd_6x_coco/ \ ...@@ -116,6 +116,18 @@ paddle2onnx --model_dir=./output_inference/rtdetr_r50vd_6x_coco/ \
--save_file rtdetr_r50vd_6x_coco.onnx --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> </details>
## 引用RT-DETR ## 引用RT-DETR
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册