pull/6/head
triple-Mu 2 years ago
parent e539828600
commit e786f4fb15
  1. 5
      README.md
  2. 0
      csrc/detect/CMakeLists.txt
  3. 0
      csrc/detect/include/config.h
  4. 0
      csrc/detect/include/utils.h
  5. 0
      csrc/detect/include/yolov8.hpp
  6. 0
      csrc/detect/main.cpp

@ -89,7 +89,8 @@ If you just want to taste first, you can download the onnx model which are expor
[**YOLOv8-x**](https://triplemu.oss-cn-beijing.aliyuncs.com/YOLOv8/ONNX/yolov8x_nms.onnx?OSSAccessKeyId=LTAI5tN1dgmZD4PF8AJUXp3J&Expires=1673936778&Signature=3o%2F7QKhiZg1dW3I6sDrY4ug6MQU%3D)
## Export Engine by TensorRT Python api
## Export TensorRT Engine
### 1. Export Engine by TensorRT Python api
You can export TensorRT engine from ONNX by [`build.py` ](build.py).
@ -116,7 +117,7 @@ python3 build.py \
You can modify `iou-thres` `conf-thres` `topk` by yourself.
## 2. Export Engine by Trtexec Tools
### 2. Export Engine by Trtexec Tools
You can export TensorRT engine by [`trtexec`](https://github.com/NVIDIA/TensorRT/tree/main/samples/trtexec) tools.

Loading…
Cancel
Save