Neural-Network-Visualization/main.cpp at main · yyxuanyao/Neural-Network-Visualization · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
#include "mainwindow.h"
#include "backend.h"
#include <QApplication>
#include "networkvisualizer.h"
#include "layerblockitem.h"

int main(int argc, char *argv[])
{
    QApplication a(argc, argv);
    /*
    MainWindow w;
    w.show();
    return a.exec();*/
    // 构造一些测试数据
    QList<NeuralLayer> layers;
    NeuralLayer l1; l1.layerType = "input"; l1.inputSize = 18; l1.neurons = 18; l1.activationFunction = " ";
    NeuralLayer l2; l2.layerType = "hidden layer"; l2.inputSize = 18; l2.neurons = 10; l2.activationFunction = "linear";
    NeuralLayer l3; l3.layerType = "hidden layer"; l3.inputSize = 18; l3.neurons = 10; l3.activationFunction = "linear";
    NeuralLayer l4; l4.layerType = "output"; l4.inputSize = 18; l4.neurons = 1; l4.activationFunction = "softmax";
    layers.append(l1);
    layers.append(l2);
    layers.append(l3);
    layers.append(l4);

    NetworkVisualizer* visualizer = new NetworkVisualizer();
    visualizer->createNetwork(layers);
    visualizer->show();
/*
    visualizer->createblockNetwork(layers);
    visualizer->show();*/

    return a.exec();
}