-
Notifications
You must be signed in to change notification settings - Fork 0
/
utils.cpp
70 lines (59 loc) · 1.38 KB
/
utils.cpp
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
#pragma once
#include <time.h>
#include <omp.h>
#define KER 7
uint64_t nanos () {
struct timespec ts;
clock_gettime(CLOCK_MONOTONIC, &ts);
return (uint64_t)ts.tv_sec*1000000000 + ts.tv_nsec;
}
void gaussian_kernel (float* kernel, float sigma) {
float sum = 0.0;
int center = KER / 2;
for (int i = 0; i < KER; ++i) {
for (int j = 0; j < KER; ++j) {
int x = i - center;
int y = j - center;
kernel[i * KER + j] = exp(-(x * x + y * y) / (2 * sigma * sigma));
sum += kernel[i * KER + j];
}
}
for (int i = 0; i < KER; ++i) {
for (int j = 0; j < KER; ++j) {
kernel[i * KER + j] /= sum;
}
}
}
void emboss_kernel (float* kernel) {
kernel[0] = -2.0;
kernel[1] = -1.0;
kernel[2] = 0.0;
kernel[3] = -1.0;
kernel[4] = 1.0;
kernel[5] = 1.0;
kernel[6] = 0.0;
kernel[7] = 1.0;
kernel[8] = 2.0;
}
void sharpen_kernel(float* kernel) {
kernel[0] = 0.0;
kernel[1] = -1.0;
kernel[2] = 0.0;
kernel[3] = -1.0;
kernel[4] = 5.0;
kernel[5] = -1.0;
kernel[6] = 0.0;
kernel[7] = -1.0;
kernel[8] = 0.0;
}
void edge_kernel(float* kernel) {
kernel[0] = 0.0;
kernel[1] = 1.0;
kernel[2] = 0.0;
kernel[3] = 1.0;
kernel[4] = -4.0;
kernel[5] = 1.0;
kernel[6] = 0.0;
kernel[7] = 1.0;
kernel[8] = 0.0;
}