Skip to content

Commit c0ae2ef

Browse files
committed
add notes
1 parent 2829a90 commit c0ae2ef

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

README.md

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -93,6 +93,11 @@ cd bin
9393
./object_detection yolov5 /path/to/input/dir
9494
```
9595

96+
> Notes:
97+
> 1. The output of the model is required for post-processing is num_bboxes (imageHeight x image Width) x num_pred(num_cls + coordinates + confidence),while the output of YOLOv8 is num_pred * num_bboxes,which means the predicted values of the same box are not contiguous in memory.For convenience, the corresponding dimensions of the original pytorch output need to be transposed when exporting to ONNX model.
98+
99+
100+
96101
## 5.Reference
97102
[0].https://github.com/NVIDIA/TensorRT<br>
98103
[1].https://docs.nvidia.com/deeplearning/tensorrt/developer-guide/index.html#c_topics<br>

0 commit comments

Comments
 (0)