diff options
Diffstat (limited to 'tests/TfResNext_Quantized-Armnn/TfResNext_Quantized-Armnn.cpp')
-rw-r--r-- | tests/TfResNext_Quantized-Armnn/TfResNext_Quantized-Armnn.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/TfResNext_Quantized-Armnn/TfResNext_Quantized-Armnn.cpp b/tests/TfResNext_Quantized-Armnn/TfResNext_Quantized-Armnn.cpp index c152c0f6c2..bec2771d4b 100644 --- a/tests/TfResNext_Quantized-Armnn/TfResNext_Quantized-Armnn.cpp +++ b/tests/TfResNext_Quantized-Armnn/TfResNext_Quantized-Armnn.cpp @@ -3,7 +3,7 @@ // SPDX-License-Identifier: MIT // #include "../InferenceTest.hpp" -#include "../CaffePreprocessor.hpp" +#include "../ImagePreprocessor.hpp" #include "armnnTfParser/ITfParser.hpp" int main(int argc, char* argv[]) @@ -21,7 +21,7 @@ int main(int argc, char* argv[]) armnn::TensorShape inputTensorShape({ 1, 3, 224, 224 }); using DataType = float; - using DatabaseType = CaffePreprocessor; + using DatabaseType = ImagePreprocessor<DataType>;; using ParserType = armnnTfParser::ITfParser; using ModelType = InferenceModel<ParserType, DataType>; |