Display training digits in UI while training

This commit is contained in:
Michael Mandl 2015-10-27 18:20:50 +01:00
parent f9be5ca717
commit 5778afa121
6 changed files with 57 additions and 7 deletions

View file

@ -3,6 +3,7 @@
#include "trainingdataloader.h"
#include <QElapsedTimer>
#include <QImage>
void NetLearner::run()
{
@ -47,6 +48,17 @@ void NetLearner::run()
{
const TrainingDataLoader::Sample &trainingSample = dataLoader.getRandomSample();
QImage sampleImage(32, 32, QImage::Format_ARGB32);
for (unsigned int y = 0; y < 32; ++y)
{
for (unsigned int x = 0; x < 32; ++x)
{
uchar grayValue = trainingSample.second[x + y * 32] * 255;
sampleImage.setPixel(x, y, qRgb(grayValue, grayValue, grayValue));
}
}
emit sampleImageLoaded(sampleImage);
std::vector<double> targetValues =
{
trainingSample.first / 10.0