Cifar10Database Class Reference

#include <Cifar10Database.hpp>

Public Types
using	DataType = float
using	TTestCaseData = ClassifierTestCaseData< DataType >

Public Member Functions
	Cifar10Database (const std::string &binaryFileDirectory, bool rgbPack=false)
std::unique_ptr< TTestCaseData >	GetTestCaseData (unsigned int testCaseId)

Detailed Description

Definition at line 12 of file Cifar10Database.hpp.

Member Typedef Documentation

DataType

using DataType = float

Definition at line 15 of file Cifar10Database.hpp.

TTestCaseData

using TTestCaseData = ClassifierTestCaseData<DataType>

Definition at line 16 of file Cifar10Database.hpp.

Constructor & Destructor Documentation

Cifar10Database()

Cifar10Database ( const std::string &  binaryFileDirectory, bool  rgbPack = false  )

explicit

Definition at line 15 of file Cifar10Database.cpp.

    : m_BinaryDirectory(binaryFileDirectory), m_RgbPack(rgbPack)
{
}

Member Function Documentation

GetTestCaseData()

std::unique_ptr< Cifar10Database::TTestCaseData > GetTestCaseData

(

unsigned int

testCaseId

)

Definition at line 20 of file Cifar10Database.cpp.

References ARMNN_LOG, g_kCifar10ImageByteSize, and armnn::numeric_cast().

Referenced by main().

{
    std::vector<unsigned char> I(g_kCifar10ImageByteSize);

    std::string fullpath = m_BinaryDirectory + std::string("test_batch.bin");

    std::ifstream fileStream(fullpath, std::ios::binary);
    if (!fileStream.is_open())
    {
        ARMNN_LOG(fatal) << "Failed to load " << fullpath;
        return nullptr;
    }

    fileStream.seekg(testCaseId * g_kCifar10ImageByteSize, std::ios_base::beg);
    fileStream.read(reinterpret_cast<char*>(&I[0]), g_kCifar10ImageByteSize);

    if (!fileStream.good())
    {
        ARMNN_LOG(fatal) << "Failed to read " << fullpath;
        return nullptr;
    }


    std::vector<float> inputImageData;
    inputImageData.resize(g_kCifar10ImageByteSize - 1);

    unsigned int step;
    unsigned int countR_o;
    unsigned int countG_o;
    unsigned int countB_o;
    unsigned int countR = 1;
    unsigned int countG = 1 + 32 * 32;
    unsigned int countB = 1 + 2 * 32 * 32;

    if (m_RgbPack)
    {
        countR_o = 0;
        countG_o = 1;
        countB_o = 2;
        step = 3;
    }
    else
    {
        countR_o = 0;
        countG_o = 32 * 32;
        countB_o = 2 * 32 * 32;
        step = 1;
    }

    for (unsigned int h = 0; h < 32; h++)
    {
        for (unsigned int w = 0; w < 32; w++)
        {
            // Static_cast of unsigned char is safe with float
            inputImageData[countR_o] = static_cast<float>(I[countR++]);
            inputImageData[countG_o] = static_cast<float>(I[countG++]);
            inputImageData[countB_o] = static_cast<float>(I[countB++]);

            countR_o += step;
            countG_o += step;
            countB_o += step;
        }
    }

    const unsigned int label = armnn::numeric_cast<unsigned int>(I[0]);
    return std::make_unique<TTestCaseData>(label, std::move(inputImageData));
}

---

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

• tests/Cifar10Database.hpp
• tests/Cifar10Database.cpp