summaryrefslogtreecommitdiff
path: root/Year_2/IandM/convolution
diff options
context:
space:
mode:
Diffstat (limited to 'Year_2/IandM/convolution')
-rw-r--r--Year_2/IandM/convolution/convolution.pde73
-rw-r--r--Year_2/IandM/convolution/data/lena.pngbin0 -> 473831 bytes
2 files changed, 73 insertions, 0 deletions
diff --git a/Year_2/IandM/convolution/convolution.pde b/Year_2/IandM/convolution/convolution.pde
new file mode 100644
index 0000000..cb2acf5
--- /dev/null
+++ b/Year_2/IandM/convolution/convolution.pde
@@ -0,0 +1,73 @@
+float n3 = 1/9.0;
+float[][] nbox3 = {{n3, n3, n3},
+ {n3, n3, n3},
+ {n3, n3, n3}};
+
+float n5 = 1.0/25;
+float[][] nbox5 = {{n5, n5, n5, n5, n5},
+ {n5, n5, n5, n5, n5},
+ {n5, n5, n5, n5, n5},
+ {n5, n5, n5, n5, n5},
+ {n5, n5, n5, n5, n5}};
+
+float[][] lapl = {{-1, 0, -1},
+ {0, 4, 0},
+ {-1, 0, -1}};
+
+float[][] sobx = {{-1, -2, -1},
+ {0, 0, 0},
+ {1, 2, 1}};
+
+float[][] sharp = {{-1, 0, -1},
+ {0, 5, 0},
+ {-1, 0, -1}};
+
+void setup() {
+ size(768, 512);
+ PImage img = loadImage("lena.png");
+ img.resize(256, 256);
+ img.filter(GRAY);
+
+ image(img, 0, 0);
+ image(convert(convolution(img, nbox3)), 256, 0);
+ image(convert(convolution(img, nbox5)), 512, 0);
+ image(convert(convolution(img, lapl)), 0, 256);
+ image(convert(convolution(img, sobx)), 256, 256);
+ image(convert(convolution(img, sharp)), 512, 256);
+}
+
+
+PImage convert(float[][] mat) {
+ PImage out = createImage(mat[0].length, mat.length, RGB);
+ float res = 0;
+
+ for (int x = 0; x < out.width; ++x) {
+ for (int y = 0; y < out.height; ++y) {
+ res = constrain(abs(mat[y][x]), 0, 255);
+ out.set(x, y, color(res));
+ }
+ }
+
+ return out;
+}
+
+float[][] convolution(PImage I, float[][] kernel) {
+ float[][] out = new float[I.height][I.width];
+ int n = kernel[0].length;
+ int m = kernel.length;
+ PImage tmp;
+
+ for (int x = 0; x < I.width; ++x) {
+ for (int y = 0; y < I.height; ++y) {
+ tmp = I.get(x-n/2, y-m/2, n, m);
+ float res = 0;
+ for (int p = 0; p < n; ++p)
+ for (int q = 0; q < m; ++q)
+ res+=red(tmp.get(p, q)) * kernel[q][p];
+
+ out[y][x] = res;
+ }
+ }
+
+ return out;
+}
diff --git a/Year_2/IandM/convolution/data/lena.png b/Year_2/IandM/convolution/data/lena.png
new file mode 100644
index 0000000..59ef68a
--- /dev/null
+++ b/Year_2/IandM/convolution/data/lena.png
Binary files differ