SRCNN-cpp/main.cpp at master · Cuda-Chen/SRCNN-cpp · 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
34
#include <iostream>

#include "srcnn.hpp"

using namespace std;

int main(int argc, char **argv)
{
    if(argc != 2)
    {
        cout << "Usage: ./SRCNN_cpp <input image path>" << endl;
        return 1;
    }
    string filename = argv[1];

    SRCNN srcnn;

    //srcnn.checkWeightStatus();
    srcnn.generate(filename);
    //srcnn.showOutput();
    //srcnn.outputImage();

    //srcnn.testImageConv(filename);
    //srcnn.testConv();
    //srcnn.testConv1Channel();
    //srcnn.testConv3Channels();
    //srcnn.testTranspose();
    //srcnn.generate(filename);
    //srcnn.showOutput();
    //srcnn.testReadAndTranspose();
    //srcnn.testReadWeightFormat();

    return 0;
}