Method readDigits

samples/sampleIOFormats/sampleIOFormats.cpp:465–490

\brief Reads the digit map from file

//! \brief Reads the digit map from file
//!
bool SampleIOFormats::readDigits(SampleBuffer& buffer, int32_t groundTruthDigit)
{
    int32_t const inputH = buffer.dims.d[2];
    int32_t const inputW = buffer.dims.d[3];

    // Read a random digit file
    std::vector<uint8_t> fileData(inputH * inputW);
    readPGMFile(
        locateFile(std::to_string(groundTruthDigit) + ".pgm", mParams.dataDirs), fileData.data(), inputH, inputW);

    // Print ASCII representation of digit
    for (int32_t i = 0; i < inputH * inputW; i++)
    {
        sample::gLogInfo << (" .:-=+*#%@"[fileData[i] / 26]) << (((i + 1) % inputW) ? "" : "\n");
    }
    sample::gLogInfo << std::endl;

    float* inputBuf = reinterpret_cast<float*>(buffer.buffer);

    for (int32_t i = 0; i < inputH * inputW; i++)
    {
        inputBuf[i] = 1.0F - static_cast<float>(fileData[i] / 255.0F);
    }

    return true;
}

//!
//! \brief Verifies that the output is correct and prints it