未验证 提交 f5bbd103 编写于 作者: G Guanghua Yu 提交者: GitHub

update yolo tensorrt docs

上级 2cabaccf
# TensorRT Python预测 # TensorRT Python预测
### 环境要求
- TensorRT 8.0以上
### 验证COCO mAP ### 验证COCO mAP
-FP16 - FP16
```shell ```shell
python trt_eval.py --onnx_model_file=yolov7_tiny_quant_onnx/yolov7-tiny.onnx \ python trt_eval.py --onnx_model_file=yolov7_tiny_quant_onnx/yolov7-tiny.onnx \
--precision_mode=fp16 \ --precision_mode=fp16 \
...@@ -38,4 +42,4 @@ python trt_eval.py --onnx_model_file=yolov7_tiny_quant_onnx/yolov7_tiny_quant.on ...@@ -38,4 +42,4 @@ python trt_eval.py --onnx_model_file=yolov7_tiny_quant_onnx/yolov7_tiny_quant.on
### FAQ ### FAQ
- 测试内存和显存占用时,首次运行会将ONNX模型转换成TRT模型,耗时不准确,再运行trt_eval.py可获取真实的内存和显存占用。 - 测试内存和显存占用时,首次运行会将ONNX模型转换成TRT模型,耗时不准确,再运行trt_eval.py可获取真实的内存和显存占用。
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册