ImagePreprocessor< TDataType > Class Template Reference

#include <ImagePreprocessor.hpp>

Public Types
enum	DataFormat { NHWC, NCHW }
using	DataType = TDataType
using	TTestCaseData = ClassifierTestCaseData< DataType >

Public Member Functions
	ImagePreprocessor (const std::string &binaryFileDirectory, unsigned int width, unsigned int height, const std::vector< ImageSet > &imageSet, 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, unsigned int batchSize=1)
std::unique_ptr< TTestCaseData >	GetTestCaseData (unsigned int testCaseId)
template<>
std::unique_ptr< ImagePreprocessor< float >::TTestCaseData >	GetTestCaseData (unsigned int testCaseId)
template<>
std::unique_ptr< ImagePreprocessor< uint8_t >::TTestCaseData >	GetTestCaseData (unsigned int testCaseId)

Detailed Description

template<typename TDataType>
class ImagePreprocessor< TDataType >

Definition at line 20 of file ImagePreprocessor.hpp.

Member Typedef Documentation

DataType

using DataType = TDataType

Definition at line 23 of file ImagePreprocessor.hpp.

TTestCaseData

using TTestCaseData = ClassifierTestCaseData<DataType>

Definition at line 24 of file ImagePreprocessor.hpp.

Member Enumeration Documentation

DataFormat

enum DataFormat

Enumerator
NHWC
NCHW

Definition at line 26 of file ImagePreprocessor.hpp.

    {
        NHWC,
        NCHW
    };

Constructor & Destructor Documentation

ImagePreprocessor()

ImagePreprocessor ( const std::string &  binaryFileDirectory, unsigned int  width, unsigned int  height, const std::vector< ImageSet > &  imageSet, 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, unsigned int  batchSize = 1  )

inlineexplicit

Definition at line 32 of file ImagePreprocessor.hpp.

References ImagePreprocessor< TDataType >::GetTestCaseData().

                                      {{0, 0, 0}},
        const std::array<float, 3> stddev={{1, 1, 1}},
        DataFormat dataFormat=DataFormat::NHWC,
        unsigned int batchSize=1)
    : m_BinaryDirectory(binaryFileDirectory)
    , m_Height(height)
    , m_Width(width)
    , m_BatchSize(batchSize)
    , m_Scale(scale)
    , m_ImageSet(imageSet)
    , m_Mean(mean)
    , m_Stddev(stddev)
    , m_DataFormat(dataFormat)
    {
    }

Member Function Documentation

GetTestCaseData() [1/3]

std::unique_ptr<TTestCaseData> GetTestCaseData

(

unsigned int

testCaseId

)

Referenced by ImagePreprocessor< TDataType >::ImagePreprocessor().

GetTestCaseData() [2/3]

std::unique_ptr< ImagePreprocessor< float >::TTestCaseData > GetTestCaseData

(

unsigned int

testCaseId

)

Definition at line 59 of file ImagePreprocessor.cpp.

{
    std::vector<float> resized;
    auto label = GetLabelAndResizedImageAsFloat(testCaseId, resized);
    return std::make_unique<TTestCaseData>(label, std::move(resized));
}

GetTestCaseData() [3/3]

std::unique_ptr< ImagePreprocessor< uint8_t >::TTestCaseData > GetTestCaseData

(

unsigned int

testCaseId

)

Definition at line 68 of file ImagePreprocessor.cpp.

{
    std::vector<float> resized;
    auto label = GetLabelAndResizedImageAsFloat(testCaseId, resized);

    size_t resizedSize = resized.size();
    std::vector<uint8_t> quantized(resized.size());

    for (size_t i=0; i<resizedSize; ++i)
    {
        quantized[i] = static_cast<uint8_t>(resized[i]);
    }

    return std::make_unique<TTestCaseData>(label, std::move(quantized));
}

---

The documentation for this class was generated from the following files:

• tests/ImagePreprocessor.hpp
• tests/ImagePreprocessor.cpp