MNISTPatchImporter.hpp

Go to the documentation of this file.

#ifndef SRC_importers_MNISTPATCHIMAGEIMPORTER_HPP_
#define SRC_importers_MNISTPATCHIMAGEIMPORTER_HPP_

#include <importers/Importer.hpp>
#include <types/MNISTTypes.hpp>

namespace mic {
namespace importers {

class MNISTPatchImporter : public mic::importers::Importer<mic::types::MatrixXf, unsigned int> {
public:
    MNISTPatchImporter(std::string node_name_ = "mnist_patch_importer");

    virtual ~MNISTPatchImporter() { };

    bool importData();

    void setDataFilename(std::string data_filename_) {
        data_filename = data_filename_;
    }

    void setLabelsFilename(std::string labels_filename_) {
        labels_filename = labels_filename_;
    }

    virtual void initializePropertyDependentVariables() { };

    size_t getPatchSize() { return patch_size; }

private:
    size_t image_width;

    size_t image_height;

    mic::configuration::Property<std::string> data_filename;

    mic::configuration::Property<std::string> labels_filename;

    mic::configuration::Property<size_t> patch_size;

    mic::configuration::Property<int> samples_limit;
};

} /* namespace importers */
} /* namespace mic */


#endif /* SRC_importers_MNISTPATCHIMAGEIMPORTER_HPP_ */