Learning_Computer_Vision_by_examples/main.py at master · SC-One/Learning_Computer_Vision_by_examples · 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
from SimpleFramework.SimpleFrameworkImageApplier import run_gui
from ExampleProcessors import PaddingProcessor, CartoonEffectProcessor, RotateProcessor, AddNoiseProcessor, GaussianBlurProcessor
from ExampleTorchProcessors import Conv2DProcessor
import numpy as np


if __name__ == "__main__":
    # 1. Sharpening
    sharpening_kernel = np.array([
        [0, -1,  0],
        [-1,  5, -1],
        [0, -1,  0]
    ], dtype=np.float32)

    # 2. Box Blur (Average)
    box_blur_kernel = np.ones((3, 3), dtype=np.float32) / 9.0

    # 3. Gaussian Blur (approximate 3×3)
    gaussian_kernel = np.array([
        [1, 2, 1],
        [2, 4, 2],
        [1, 2, 1]
    ], dtype=np.float32) / 16.0

    # 4. Horizontal Edge Detection (Sobel X)
    sobel_x_kernel = np.array([
        [-1, 0, 1],
        [-2, 0, 2],
        [-1, 0, 1]
    ], dtype=np.float32)

    # 5. Vertical Edge Detection (Sobel Y)
    sobel_y_kernel = np.array([
        [-1, -2, -1],
        [0,  0,  0],
        [1,  2,  1]
    ], dtype=np.float32)

    # 6. Laplacian (Edge Detection)
    laplacian_kernel = np.array([
        [0,  1,  0],
        [1, -4,  1],
        [0,  1,  0]
    ], dtype=np.float32)

    # 7. Emboss
    emboss_kernel = np.array([
        [-2, -1,  0],
        [-1,  1,  1],
        [0,  1,  2]
    ], dtype=np.float32)

    run_gui("example.png", [PaddingProcessor(),
                            CartoonEffectProcessor(),
                            RotateProcessor(),
                            AddNoiseProcessor(),
                            GaussianBlurProcessor(),
                            Conv2DProcessor("(Conv2d sharpening_kernel) maxpool BN", sharpening_kernel),
                            Conv2DProcessor("(Conv2d box_blur_kernel) maxpool BN", box_blur_kernel),
                            Conv2DProcessor("(Conv2d gaussian_kernel) maxpool BN", gaussian_kernel),
                            Conv2DProcessor("(Conv2d sobel_x_kernel) maxpool BN", sobel_x_kernel),
                            Conv2DProcessor("(Conv2d sobel_y_kernel) maxpool BN", sobel_y_kernel),
                            Conv2DProcessor("(Conv2d laplacian_kernel) maxpool BN", laplacian_kernel),
                            Conv2DProcessor("(Conv2d emboss_kernel) maxpool BN", emboss_kernel) ])