diff options
author | James Conroy <james.conroy@arm.com> | 2018-11-26 09:48:30 +0000 |
---|---|---|
committer | James Conroy <james.conroy@arm.com> | 2018-11-26 11:16:32 +0000 |
commit | e02d60154d8230f555c832c1cce1865b54dc6589 (patch) | |
tree | c66d63184de3e9d1850e8bb15e1cdfcb886faab1 /tests/YoloDatabase.cpp | |
parent | 4b19381b6ddaff97bec0e1e7225f8e95c43db42c (diff) | |
download | armnn-e02d60154d8230f555c832c1cce1865b54dc6589.tar.gz |
IVGCVSW-2199 Call Resize only if H and/or W changes
* Wrap Resize function call with conditional checking
if height and/or width need to be changed.
Change-Id: Ic33c3ed29cc985c147d29db8e9dd237ab9c741d1
Diffstat (limited to 'tests/YoloDatabase.cpp')
-rw-r--r-- | tests/YoloDatabase.cpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tests/YoloDatabase.cpp b/tests/YoloDatabase.cpp index 0f5dee026c..472fc8083d 100644 --- a/tests/YoloDatabase.cpp +++ b/tests/YoloDatabase.cpp @@ -83,7 +83,10 @@ std::unique_ptr<YoloDatabase::TTestCaseData> YoloDatabase::GetTestCaseData(unsig try { InferenceTestImage image(imagePath.c_str()); - image.Resize(YoloImageWidth, YoloImageHeight, CHECK_LOCATION()); + if (YoloImageWidth != image.GetWidth() || YoloImageHeight != image.GetHeight()) + { + image.Resize(YoloImageWidth, YoloImageHeight, CHECK_LOCATION()); + } imageData = GetImageDataInArmNnLayoutAsNormalizedFloats(ImageChannelLayout::Rgb, image); } catch (const InferenceTestImageException& e) |