STL10MatrixImporter.hpp

Go to the documentation of this file.

#ifndef SRC_importers_STL10MatrixImporter_HPP_
#define SRC_importers_STL10MatrixImporter_HPP_

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

namespace mic {
namespace importers {

class STL10MatrixImporter: public mic::importers::Importer< mic::types::MatrixXf, unsigned int > {
public:
    STL10MatrixImporter(std::string node_name_ = "stl10_matrix_importer", std::string data_filename_ = "", std::string labels_filename_ = "");

    virtual ~STL10MatrixImporter() { };

    void setDataFilename(std::string data_filename_);

    void setLabelsFilename(std::string labels_filename_);

    bool importData();

    virtual void initializePropertyDependentVariables() { };

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<int> samples_limit;

    bool labeled = false;
};


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


#endif /* SRC_importers_STL10MatrixImporter_HPP_ */