我正在尝试构建并运行tensorrt MNIST样本程序(https://github.com/NVIDIA/TensorRT/tree/release/6.0/samples/opensource/sampleMNIST),但是当我运行它时,该推理失败了,因为它总是预测相同的输出(数字8)。
这里显示了如何运行应用程序以及获得的输出。
谢谢, 埃斯特万
poncos@electron:~/CLionProjects/tensorrt$ ./bin/sampleMNIST --datadir=./resources
&&&& RUNNING TensorRT.sample_mnist # ./bin/sampleMNIST --datadir=./resources
[08/25/2019-23:57:46] [I] Building and running a GPU inference engine for MNIST
[08/25/2019-23:57:47] [I] [TRT] Detected 1 inputs and 1 output network tensors.
[08/25/2019-23:57:47] [I] Input:
--:.:===-=.--=====.... .:..:
.::
@@@
@@@@@
@@@@@@@
@ @@@@@ @@@
@@@@@@@@ @@@
@@@@@ @@@
@@@@ @@@
@@@ @@@
@ @@@
@@@
@@@
@@@
@@@
@@@
@@@
@@@
@@@
@@@
@
[08/25/2019-23:57:47] [I] Output:
0:
1:
2:
3:
4:
5:
6:
7:
8: **********
9:
&&&& FAILED TensorRT.sample_mnist # ./bin/sampleMNIST --datadir=./resources
poncos@electron:~/CLionProjects/tensorrt$