-
Notifications
You must be signed in to change notification settings - Fork 42
Description
如题,不知道作者是否遇到过,训练完成的yolov5x模型,在python版本正确率为98%,但是转换为tensorrt后经过测试,正确率只有90%左右,其中模型转换过程log如下:
[09/27/2021-11:27:04] [I] Host Latency
[09/27/2021-11:27:04] [I] min: 11.3848 ms (end to end 21.3677 ms)
[09/27/2021-11:27:04] [I] max: 13.1256 ms (end to end 24.1753 ms)
[09/27/2021-11:27:04] [I] mean: 11.67 ms (end to end 21.9034 ms)
[09/27/2021-11:27:04] [I] median: 11.5836 ms (end to end 21.7285 ms)
[09/27/2021-11:27:04] [I] percentile: 12.5283 ms at 99% (end to end 23.6667 ms at 99%)
[09/27/2021-11:27:04] [I] throughput: 0 qps
[09/27/2021-11:27:04] [I] walltime: 3.03151 s
[09/27/2021-11:27:04] [I] Enqueue Time
[09/27/2021-11:27:04] [I] min: 1.04535 ms
[09/27/2021-11:27:04] [I] max: 4.6637 ms
[09/27/2021-11:27:04] [I] median: 1.61969 ms
[09/27/2021-11:27:04] [I] GPU Compute
[09/27/2021-11:27:04] [I] min: 10.8311 ms
[09/27/2021-11:27:04] [I] max: 12.5458 ms
[09/27/2021-11:27:04] [I] mean: 11.0955 ms
[09/27/2021-11:27:04] [I] median: 11.0142 ms
[09/27/2021-11:27:04] [I] percentile: 11.9821 ms at 99%
[09/27/2021-11:27:04] [I] total compute time: 3.01798 s
&&&& PASSED TensorRT.trtexec # trtexec.exe --onnx=best.onnx --saveEngine=best.engine --fp16