CoCalc -- kernels.py

📚 The CoCalc Library - books, templates and other resources
cocalc-examples / stanford-tensorflow-tutorials / examples / kernels.py
¹³²⁹²³ views
License: OTHER
1
import numpy as np
2
import tensorflow as tf
3

4
a = np.zeros([3, 3, 3, 3])
5
a[1, 1, :, :] = 0.25
6
a[0, 1, :, :] = 0.125
7
a[1, 0, :, :] = 0.125
8
a[2, 1, :, :] = 0.125
9
a[1, 2, :, :] = 0.125
10
a[0, 0, :, :] = 0.0625
11
a[0, 2, :, :] = 0.0625
12
a[2, 0, :, :] = 0.0625
13
a[2, 2, :, :] = 0.0625
14

15
BLUR_FILTER_RGB = tf.constant(a, dtype=tf.float32)
16

17
a = np.zeros([3, 3, 1, 1])
18
# a[1, 1, :, :] = 0.25
19
# a[0, 1, :, :] = 0.125
20
# a[1, 0, :, :] = 0.125
21
# a[2, 1, :, :] = 0.125
22
# a[1, 2, :, :] = 0.125
23
# a[0, 0, :, :] = 0.0625
24
# a[0, 2, :, :] = 0.0625
25
# a[2, 0, :, :] = 0.0625
26
# a[2, 2, :, :] = 0.0625
27
a[1, 1, :, :] = 1.0
28
a[0, 1, :, :] = 1.0
29
a[1, 0, :, :] = 1.0
30
a[2, 1, :, :] = 1.0
31
a[1, 2, :, :] = 1.0
32
a[0, 0, :, :] = 1.0
33
a[0, 2, :, :] = 1.0
34
a[2, 0, :, :] = 1.0
35
a[2, 2, :, :] = 1.0
36
BLUR_FILTER = tf.constant(a, dtype=tf.float32)
37

38
a = np.zeros([3, 3, 3, 3])
39
a[1, 1, :, :] = 5
40
a[0, 1, :, :] = -1
41
a[1, 0, :, :] = -1
42
a[2, 1, :, :] = -1
43
a[1, 2, :, :] = -1
44

45
SHARPEN_FILTER_RGB = tf.constant(a, dtype=tf.float32)
46

47
a = np.zeros([3, 3, 1, 1])
48
a[1, 1, :, :] = 5
49
a[0, 1, :, :] = -1
50
a[1, 0, :, :] = -1
51
a[2, 1, :, :] = -1
52
a[1, 2, :, :] = -1
53

54
SHARPEN_FILTER = tf.constant(a, dtype=tf.float32)
55

56
# a = np.zeros([3, 3, 3, 3])
57
# a[:, :, :, :] = -1
58
# a[1, 1, :, :] = 8
59

60
# EDGE_FILTER_RGB = tf.constant(a, dtype=tf.float32)
61

62
EDGE_FILTER_RGB = tf.constant([
63
			[[[ -1., 0., 0.], [ 0., -1., 0.], [ 0., 0., -1.]],
64
            [[ -1., 0., 0.], [ 0., -1., 0.], [ 0., 0., -1.]],
65
            [[ -1., 0., 0.], [ 0., -1., 0.], [ 0., 0., -1.]]],
66
            [[[ -1., 0., 0.], [ 0., -1., 0.], [ 0., 0., -1.]],
67
			[[ 8., 0., 0.], [ 0., 8., 0.], [ 0., 0., 8.]],
68
			[[ -1., 0., 0.], [ 0., -1., 0.], [ 0., 0., -1.]]],
69
			[[[ -1., 0., 0.], [ 0., -1., 0.], [ 0., 0., -1.]],
70
			[[ -1., 0., 0.], [ 0., -1., 0.], [ 0., 0., -1.]],
71
			[[ -1., 0., 0.], [ 0., -1., 0.], [ 0., 0., -1.]]]
72
])
73

74
a = np.zeros([3, 3, 1, 1])
75
# a[:, :, :, :] = -1
76
# a[1, 1, :, :] = 8
77
a[0, 1, :, :] = -1
78
a[1, 0, :, :] = -1
79
a[1, 2, :, :] = -1
80
a[2, 1, :, :] = -1
81
a[1, 1, :, :] = 4
82

83
EDGE_FILTER = tf.constant(a, dtype=tf.float32)
84

85
a = np.zeros([3, 3, 3, 3])
86
a[0, :, :, :] = 1
87
a[0, 1, :, :] = 2 # originally 2
88
a[2, :, :, :] = -1
89
a[2, 1, :, :] = -2
90

91
TOP_SOBEL_RGB = tf.constant(a, dtype=tf.float32)
92

93
a = np.zeros([3, 3, 1, 1])
94
a[0, :, :, :] = 1
95
a[0, 1, :, :] = 2 # originally 2
96
a[2, :, :, :] = -1
97
a[2, 1, :, :] = -2
98

99
TOP_SOBEL = tf.constant(a, dtype=tf.float32)
100

101
a = np.zeros([3, 3, 3, 3])
102
a[0, 0, :, :] = -2
103
a[0, 1, :, :] = -1 
104
a[1, 0, :, :] = -1
105
a[1, 1, :, :] = 1
106
a[1, 2, :, :] = 1
107
a[2, 1, :, :] = 1
108
a[2, 2, :, :] = 2
109

110
EMBOSS_FILTER_RGB = tf.constant(a, dtype=tf.float32)
111

112
a = np.zeros([3, 3, 1, 1])
113
a[0, 0, :, :] = -2
114
a[0, 1, :, :] = -1 
115
a[1, 0, :, :] = -1
116
a[1, 1, :, :] = 1
117
a[1, 2, :, :] = 1
118
a[2, 1, :, :] = 1
119
a[2, 2, :, :] = 2
120
EMBOSS_FILTER = tf.constant(a, dtype=tf.float32)
121
Product

Resources

Company
