BlurEdgeDetection/EdgeDetectionContinuous.cpp at master · iguha94/BlurEdgeDetection · 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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
#include "CImg.h"
#include "Global.h"
#include "Commons.h"
#include "ComputeGradient.h"
#include "continuousscaleComputation.h"
#include "MathFunctions.h"
#include <iostream>
#include <string>
#include <time.h>

using namespace std;
using namespace cimg_library;


int main(int argc, char* argv[]) {

    //string logfile=basepath+"log.txt";
    //logger.open(logfile.c_str());
    //confidence=1.5*pow(10,-8);
    //memconf=confidence;
    bool b=false;
    gaussvar=1;
    int kernelsize=3;
    sampling_rate=6;

    string curimagefile(argv[1]);
    gaussgradvar=atof(argv[2]);
    colorgaussvar=atof(argv[3]);
    maxscale=atof(argv[4]);
    snr=atof(argv[5]);
    basepath=string(argv[6]);

    basesig=gaussgradvar;
    //cout<<"Enter membership value: ";
    //cin>>memconf;

    //cout<<"Enter the pixel of interest: ";
    //cin>>Xc>>Yc;

    //cout<<"Interrmediate Scale: ";
    //cin>>intermediatescale;

    //string sfname=basepath+"gradMap.txt";
    //sout.open(sfname.c_str());

    no_of_interpolating_points=3; //denotes the number of samples taken for each point, r,r-0.25,r+0.25

    imagefilename=basepath+curimagefile;
    scaleimagename=basepath+"scale-"+curimagefile;
    gradientimagename=basepath+"scale-gradient-"+curimagefile;
    gaussgradientimagename=basepath+"point-gradient-"+curimagefile;
    sordergradientimagename=basepath+"sogradient-"+curimagefile;
    sorderscaleimagename=basepath+"soscale-"+curimagefile;
    gradientcolorimagename=basepath+"scale-gradientcolor-"+curimagefile;
    gausscolorimagename=basepath+"point-gradient-color-"+curimagefile;
    localmaximaimagename=basepath+"localmaxima-"+curimagefile;
    softlocalmaximaimagename=basepath+"softlocalmaxima-"+curimagefile;
    Hysteresisimagename=basepath+"scalebasededge_"+curimagefile;
    secondderivativeimagename=basepath+"sd-"+curimagefile;
    localizedimagename=basepath+"Canny-"+curimagefile;
    dialatedimagename=basepath+"dialated-"+curimagefile;
    dividorimagename=basepath+"dividor-"+curimagefile;

    CImg<unsigned short> jpgimage(imagefilename.c_str());
    Xdim=jpgimage.width();
    Ydim=jpgimage.height();
    cout<<"Image Size: "<<Xdim<< " X "<<Ydim<<"\n";
    allocateMemory(Xdim,Ydim);
    allocateGradientImageMemory(Xdim,Ydim);
    maxkernelsize=7*maxscale;
    if(maxkernelsize%2==0) maxkernelsize+=1;
    allocateKernelMemory(maxscale,maxkernelsize);
    PrecomputeKernels(maxscale);
    cout<<"Kernels Precomputed\n";

    readImage<unsigned short>(jpgimage,image);

    noisestdv=snr*snr;
    gradienthresh=6;
    gaussgradvar=atof(argv[2]);
    kernelsize=6*gaussgradvar;
    if(kernelsize%2==0) kernelsize+=1;
    kernelsize=kernelsize>3?kernelsize:3;
    initializeKernels(kernelsize);
    cout<<"Initilaized Point Gradients\n";
    compute_Gradient(image,Xdim,Ydim);
    compute_scale_Gradient_with_continuous_interpolation(image,Xdim,Ydim,b);
    sout.close();
    logger.close();

    return 0;
}