Образец проекта Tensorrt wit MNIST не проходит проверку правильности - PullRequest
0 голосов
/ 26 сентября 2019

Я пытаюсь построить и запустить пример проекта tenorrt 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$ 
...