diff options
author | FinnWilliamsArm <Finn.Williams@arm.com> | 2019-05-22 14:50:55 +0100 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-05-28 09:59:44 +0100 |
commit | a723ec5d2ac35948efb5dfd0c121a1a89cb977b7 (patch) | |
tree | e0a101fbb7ba41cd50e30a1e0ce31529404175ed /tests/ImagePreprocessor.hpp | |
parent | 7f2c35a82ec11be50b3478bd15207320bbf3bd57 (diff) | |
download | armnn-a723ec5d2ac35948efb5dfd0c121a1a89cb977b7.tar.gz |
IVGCVSW-3129 Image pre-processing fix for TFLitev19.05
* Resized images for quantized models are now statically cast to uint8
instead of quantized
* Removed optional quantization parameters from ImagePreprocessor
constructor
* Changed mean and scale for TFLite models
Signed-off-by: FinnWilliamsArm <Finn.Williams@arm.com>
Change-Id: Id5ffdf77f3614d10c417e769bd8ffc4a4c07308b
Diffstat (limited to 'tests/ImagePreprocessor.hpp')
-rw-r--r-- | tests/ImagePreprocessor.hpp | 5 |
1 files changed, 1 insertions, 4 deletions
diff --git a/tests/ImagePreprocessor.hpp b/tests/ImagePreprocessor.hpp index d77113c6d9..cd586818c2 100644 --- a/tests/ImagePreprocessor.hpp +++ b/tests/ImagePreprocessor.hpp @@ -33,8 +33,7 @@ public: unsigned int width, unsigned int height, const std::vector<ImageSet>& imageSet, - float scale=1.0, - int32_t offset=0, + float scale=255.0f, const std::array<float, 3> mean={{0, 0, 0}}, const std::array<float, 3> stddev={{1, 1, 1}}, DataFormat dataFormat=DataFormat::NHWC, @@ -44,7 +43,6 @@ public: , m_Width(width) , m_BatchSize(batchSize) , m_Scale(scale) - , m_Offset(offset) , m_ImageSet(imageSet) , m_Mean(mean) , m_Stddev(stddev) @@ -66,7 +64,6 @@ private: unsigned int m_BatchSize; // Quantization parameters float m_Scale; - int32_t m_Offset; const std::vector<ImageSet> m_ImageSet; const std::array<float, 3> m_Mean; |