swh:1:snp:f50ab94432af916b5fb8b4ad831e8dddded77084
Raw File
Tip revision: 92738882d1e86c75b75fd08bf5c8872f9e94d697 authored by Cha Zhang on 14 August 2017, 15:46:35 UTC
Create a new Pretrained model folder.
Tip revision: 9273888
ImageReader.h
//
// Copyright (c) Microsoft. All rights reserved.
// Licensed under the MIT license. See LICENSE.md file in the project root for full license information.
//

#pragma once

#include "ReaderBase.h"
#include "SequenceEnumerator.h"

namespace CNTK {

using namespace Microsoft::MSR::CNTK;

// Implementation of the image reader.
// Effectively the class represents a factory for connecting the packer,
// transformers and deserialzier together.
class ImageReader : public ReaderBase
{
public:
    ImageReader(const ConfigParameters& parameters);

    // Description of streams that this reader provides.
    std::vector<StreamInformation> GetStreamDescriptions() override;

private:
    // All streams this reader provides.
    std::vector<StreamInformation> m_streams;

    // Seed for the random generator.
    unsigned int m_seed;
};

}
back to top