2015-10-26 21:05:50 +00:00
|
|
|
#ifndef TRAININGDATALOADER_H
|
|
|
|
#define TRAININGDATALOADER_H
|
|
|
|
|
|
|
|
#include <utility>
|
|
|
|
#include <list>
|
|
|
|
#include <string>
|
|
|
|
|
|
|
|
#include <QString>
|
|
|
|
|
|
|
|
class TrainingDataLoader
|
|
|
|
{
|
|
|
|
public:
|
2015-10-27 14:33:54 +00:00
|
|
|
using SampleData = double[32*32];
|
2015-10-26 21:05:50 +00:00
|
|
|
using SampleId = unsigned int;
|
|
|
|
using Sample = std::pair<SampleId, SampleData>;
|
|
|
|
|
|
|
|
private:
|
|
|
|
std::list<Sample> m_samples;
|
|
|
|
|
|
|
|
public:
|
|
|
|
TrainingDataLoader();
|
|
|
|
|
|
|
|
void addSamples(const QString &sourceFile, SampleId sampleId);
|
2015-10-27 14:33:54 +00:00
|
|
|
|
|
|
|
const Sample &getRandomSample() const;
|
2015-10-26 21:05:50 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
#endif // TRAININGDATALOADER_H
|