22.05
|
#include "InferenceTestImage.hpp"
#include <armnn/utility/Assert.hpp>
#include <armnn/utility/IgnoreUnused.hpp>
#include <armnn/utility/NumericCast.hpp>
#include <fmt/format.h>
#include <array>
#include <stb/stb_image.h>
#include <stb/stb_image_resize.h>
#include <stb/stb_image_write.h>
Go to the source code of this file.
Macros | |
#define | STB_IMAGE_IMPLEMENTATION |
#define | STB_IMAGE_RESIZE_IMPLEMENTATION |
#define | STB_IMAGE_WRITE_IMPLEMENTATION |
Functions | |
template<typename TProcessValueCallable > | |
std::vector< float > | GetImageDataInArmNnLayoutAsFloats (ImageChannelLayout channelLayout, const InferenceTestImage &image, TProcessValueCallable processValue) |
std::vector< float > | GetImageDataInArmNnLayoutAsNormalizedFloats (ImageChannelLayout layout, const InferenceTestImage &image) |
std::vector< float > | GetImageDataInArmNnLayoutAsFloatsSubtractingMean (ImageChannelLayout layout, const InferenceTestImage &image, const std::array< float, 3 > &mean) |
std::vector< float > | GetImageDataAsNormalizedFloats (ImageChannelLayout layout, const InferenceTestImage &image) |
#define STB_IMAGE_IMPLEMENTATION |
Definition at line 15 of file InferenceTestImage.cpp.
#define STB_IMAGE_RESIZE_IMPLEMENTATION |
Definition at line 18 of file InferenceTestImage.cpp.
#define STB_IMAGE_WRITE_IMPLEMENTATION |
Definition at line 21 of file InferenceTestImage.cpp.
std::vector<float> GetImageDataAsNormalizedFloats | ( | ImageChannelLayout | layout, |
const InferenceTestImage & | image | ||
) |
Definition at line 333 of file InferenceTestImage.cpp.
References B, G, InferenceTestImage::GetHeight(), InferenceTestImage::GetPixelAs3Channels(), InferenceTestImage::GetWidth(), and R.
std::vector<float> GetImageDataInArmNnLayoutAsFloats | ( | ImageChannelLayout | channelLayout, |
const InferenceTestImage & | image, | ||
TProcessValueCallable | processValue | ||
) |
Definition at line 279 of file InferenceTestImage.cpp.
References B, G, InferenceTestImage::GetHeight(), InferenceTestImage::GetPixelAs3Channels(), InferenceTestImage::GetWidth(), and R.
Referenced by GetImageDataInArmNnLayoutAsFloatsSubtractingMean(), and GetImageDataInArmNnLayoutAsNormalizedFloats().
std::vector<float> GetImageDataInArmNnLayoutAsFloatsSubtractingMean | ( | ImageChannelLayout | layout, |
const InferenceTestImage & | image, | ||
const std::array< float, 3 > & | mean | ||
) |
Definition at line 321 of file InferenceTestImage.cpp.
References GetImageDataInArmNnLayoutAsFloats().
std::vector<float> GetImageDataInArmNnLayoutAsNormalizedFloats | ( | ImageChannelLayout | layout, |
const InferenceTestImage & | image | ||
) |
Definition at line 310 of file InferenceTestImage.cpp.
References GetImageDataInArmNnLayoutAsFloats(), and armnn::IgnoreUnused().
Referenced by YoloDatabase::GetTestCaseData().