Tensorrt机智MNIST示例程序不会通过推理验证

时间:2019-09-25 22:00:51

标签: mnist tensorrt

我正在尝试构建并运行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$ 

0 个答案:

没有答案