TensorRT sampleINT8API Demo низкая точность - PullRequest
Новая
       17

TensorRT sampleINT8API Demo низкая точность

0 голосов
/ 01 апреля 2020

Среда 

TensorRT Version: 7.0
GPU Type: T4
Nvidia Driver Version: 410.79
CUDA Version: 10.0
CUDNN Version: 7.6.4
Operating System + Version: CentOS 7.6.1810
Python Version: 2.7
TensorFlow Version: 1.15

Я запускаю TensorRT Demo sampleINT8API: https://github.com/NVIDIA/TensorRT/tree/master/samples/opensource/, и я получаю правильный результат с образцом изображения airliner.ppm: 

[I] sampleINT8API result: Detected:
[I] [1] airliner
[I] [2] warplane
[I] [3] space shuttle
[I] [4] projectile
[I] [5] wing

и я изменить размер всего набора проверки изображения _net (50000 изображений) с размерами 224x224x3 и преобразовать их в расширение PPM демо Согласно наблюдениям у c. Но после того, как я запускаю весь набор проверки, точность top-1 / top5 составляет всего 53,61% и 78,42%

Затем я попытался сначала изменить размер более короткого края каждого изображения до 244, а затем обрезать центр 244x244x3 и преобразовать их в расширение PPM, я получаю 61,63% / 84,8%. Я также стараюсь как https://github.com/onnx/models/blob/master/vision/classification/imagenet_inference.ipynb, где дает ре snet модель демо использует, они изменить размер изображения до размера 256х256, возьмите центр урожай размером 224x224, однако я получить плохой результат.

1016 * Кто-нибудь знает, как получить правильный результат?
...