diff options
Diffstat (limited to 'tests/YoloDatabase.cpp')
-rw-r--r-- | tests/YoloDatabase.cpp | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tests/YoloDatabase.cpp b/tests/YoloDatabase.cpp index 4c91384073..71362b2218 100644 --- a/tests/YoloDatabase.cpp +++ b/tests/YoloDatabase.cpp @@ -78,12 +78,12 @@ std::unique_ptr<YoloDatabase::TTestCaseData> YoloDatabase::GetTestCaseData(unsig const auto& testCaseInputOutput = g_PerTestCaseInputOutput[testCaseId]; const std::string imagePath = m_ImageDir + testCaseInputOutput.first; - // Load test case input image + // Loads test case input image. std::vector<float> imageData; try { InferenceTestImage image(imagePath.c_str()); - image.Resize(YoloImageWidth, YoloImageHeight); + image.Resize(YoloImageWidth, YoloImageHeight, CHECK_LOCATION()); imageData = GetImageDataInArmNnLayoutAsNormalizedFloats(ImageChannelLayout::Rgb, image); } catch (const InferenceTestImageException& e) @@ -92,10 +92,10 @@ std::unique_ptr<YoloDatabase::TTestCaseData> YoloDatabase::GetTestCaseData(unsig return nullptr; } - // Prepare test case output + // Prepares test case output. std::vector<YoloDetectedObject> topObjectDetections; topObjectDetections.reserve(1); topObjectDetections.push_back(testCaseInputOutput.second); return std::make_unique<YoloTestCaseData>(std::move(imageData), std::move(topObjectDetections)); -}
\ No newline at end of file +} |