main.cpp

#include "network.h"
#include "image.h"

#define  FRAME_W     2
#define  FRAME_H     2
#define  L           4

int main(int argc, char **argv)
{
    std::string inputFile(argc > 1 ? argv[1] : "example.png");
    std::string outputFile(argc > 2 ? argv[2] : "output.png");

    Png img(inputFile);
    Dataframe dataframe = img.split(FRAME_W, FRAME_H);

    auto dataset = convert(dataframe);
    auto outputSize = dataset.front().y.size();

    TrainParams params;
    params.dataset = dataset;
    params.epochs = 10;
    params.error = 1e-10;
    params.teach = 0.05;

    Model model;

    model.add(Layer(L, linear));                    // define hidden layer
    model.add(Layer(outputSize, linear));           // define output layer

    model.fit(params);

    auto testingSet = img.makeTestingSet(5, outputSize);
    model.testing(testingSet);

    for (Series &series : dataframe) {
        for (Frame &frame : series) {
            auto output = model.predict(frame.toVector());
            frame.fromVector(output);
        }
    }

    img.assemble(dataframe);
    img.dump(outputFile);

    return 0;
}