Padzong commited on
Commit
a58f3f2
·
verified ·
1 Parent(s): a42c049

Add gitignore, data, model class

Browse files
Files changed (8) hide show
  1. .gitignore +160 -0
  2. data/14_5.png +0 -0
  3. data/14_7.png +0 -0
  4. data/22_2.png +0 -0
  5. data/22_7.png +0 -0
  6. data/24_1.png +0 -0
  7. data/24_3.png +0 -0
  8. siamese_nn.py +83 -0
.gitignore ADDED
@@ -0,0 +1,160 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Byte-compiled / optimized / DLL files
2
+ __pycache__/
3
+ *.py[cod]
4
+ *$py.class
5
+
6
+ # C extensions
7
+ *.so
8
+
9
+ # Distribution / packaging
10
+ .Python
11
+ build/
12
+ develop-eggs/
13
+ dist/
14
+ downloads/
15
+ eggs/
16
+ .eggs/
17
+ lib/
18
+ lib64/
19
+ parts/
20
+ sdist/
21
+ var/
22
+ wheels/
23
+ share/python-wheels/
24
+ *.egg-info/
25
+ .installed.cfg
26
+ *.egg
27
+ MANIFEST
28
+
29
+ # PyInstaller
30
+ # Usually these files are written by a python script from a template
31
+ # before PyInstaller builds the exe, so as to inject date/other infos into it.
32
+ *.manifest
33
+ *.spec
34
+
35
+ # Installer logs
36
+ pip-log.txt
37
+ pip-delete-this-directory.txt
38
+
39
+ # Unit test / coverage reports
40
+ htmlcov/
41
+ .tox/
42
+ .nox/
43
+ .coverage
44
+ .coverage.*
45
+ .cache
46
+ nosetests.xml
47
+ coverage.xml
48
+ *.cover
49
+ *.py,cover
50
+ .hypothesis/
51
+ .pytest_cache/
52
+ cover/
53
+
54
+ # Translations
55
+ *.mo
56
+ *.pot
57
+
58
+ # Django stuff:
59
+ *.log
60
+ local_settings.py
61
+ db.sqlite3
62
+ db.sqlite3-journal
63
+
64
+ # Flask stuff:
65
+ instance/
66
+ .webassets-cache
67
+
68
+ # Scrapy stuff:
69
+ .scrapy
70
+
71
+ # Sphinx documentation
72
+ docs/_build/
73
+
74
+ # PyBuilder
75
+ .pybuilder/
76
+ target/
77
+
78
+ # Jupyter Notebook
79
+ .ipynb_checkpoints
80
+
81
+ # IPython
82
+ profile_default/
83
+ ipython_config.py
84
+
85
+ # pyenv
86
+ # For a library or package, you might want to ignore these files since the code is
87
+ # intended to run in multiple environments; otherwise, check them in:
88
+ # .python-version
89
+
90
+ # pipenv
91
+ # According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
92
+ # However, in case of collaboration, if having platform-specific dependencies or dependencies
93
+ # having no cross-platform support, pipenv may install dependencies that don't work, or not
94
+ # install all needed dependencies.
95
+ #Pipfile.lock
96
+
97
+ # poetry
98
+ # Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
99
+ # This is especially recommended for binary packages to ensure reproducibility, and is more
100
+ # commonly ignored for libraries.
101
+ # https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
102
+ #poetry.lock
103
+
104
+ # pdm
105
+ # Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
106
+ #pdm.lock
107
+ # pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
108
+ # in version control.
109
+ # https://pdm.fming.dev/#use-with-ide
110
+ .pdm.toml
111
+
112
+ # PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
113
+ __pypackages__/
114
+
115
+ # Celery stuff
116
+ celerybeat-schedule
117
+ celerybeat.pid
118
+
119
+ # SageMath parsed files
120
+ *.sage.py
121
+
122
+ # Environments
123
+ .env
124
+ .venv
125
+ env/
126
+ venv/
127
+ ENV/
128
+ env.bak/
129
+ venv.bak/
130
+
131
+ # Spyder project settings
132
+ .spyderproject
133
+ .spyproject
134
+
135
+ # Rope project settings
136
+ .ropeproject
137
+
138
+ # mkdocs documentation
139
+ /site
140
+
141
+ # mypy
142
+ .mypy_cache/
143
+ .dmypy.json
144
+ dmypy.json
145
+
146
+ # Pyre type checker
147
+ .pyre/
148
+
149
+ # pytype static type analyzer
150
+ .pytype/
151
+
152
+ # Cython debug symbols
153
+ cython_debug/
154
+
155
+ # PyCharm
156
+ # JetBrains specific template is maintained in a separate JetBrains.gitignore that can
157
+ # be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
158
+ # and can be added to the global gitignore or merged into this file. For a more nuclear
159
+ # option (not recommended) you can uncomment the following to ignore the entire idea folder.
160
+ #.idea/
data/14_5.png ADDED
data/14_7.png ADDED
data/22_2.png ADDED
data/22_7.png ADDED
data/24_1.png ADDED
data/24_3.png ADDED
siamese_nn.py ADDED
@@ -0,0 +1,83 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #%%
2
+ """
3
+ This module is responsible for architecturally defining Siamese Neural Network using PyTorch Framework.
4
+
5
+ Module consists of class definition for mentioned Neural Network.
6
+
7
+ Siamese Neural Network bases its functionality on having two images processed at the same time to produce
8
+ two vectors of similarity which then in turn are being passed to Contrastive Loss function, which in the end
9
+ indicates whether or not the samples were similar or dissimilar.
10
+ """
11
+ import torch.nn as nn
12
+ from torchsummary import summary
13
+
14
+ class Siamese_nn(nn.Module):
15
+ """
16
+ Siamese_nn class inherits nn.Module from PyTorch framework to create Sequential model.
17
+
18
+ Class consists of following methods:
19
+
20
+ - __init__(): responsible for creating featureExtractor that retrieves fingerprint features and
21
+ fc which has Fully Connected layer to create similarity vectors,
22
+ - forwardOne(x): responsible for single action of forward pass in neural network,
23
+ - forward(input1, input2): responsible for forward pass of two samples at the same time, due to nature of
24
+ Siamese Network.
25
+ """
26
+
27
+ def __init__(self):
28
+ """
29
+ Initializes two important components of Siamese Neural Network:
30
+ - featureExtractor: responsible for analysing input images to extract features of fingerprint,
31
+ it is made of several Convolutional Layers with rising amount of features
32
+ extracted for increased precision,
33
+ - fc: responsible for receiving vector of features and shrinks them to reasonably sized vector.
34
+ """
35
+ super(Siamese_nn, self).__init__()
36
+
37
+ self.featureExtractor = nn.Sequential(
38
+ nn.Conv2d(1, 32, kernel_size=5),
39
+ nn.ReLU(inplace=True),
40
+ nn.Conv2d(32, 64, kernel_size=5),
41
+ nn.ReLU(inplace=True),
42
+ nn.MaxPool2d(kernel_size=2, stride=2),
43
+ nn.Conv2d(64, 128, kernel_size=3),
44
+ nn.ReLU(inplace=True),
45
+ nn.Conv2d(128, 128, kernel_size=3),
46
+ nn.ReLU(inplace=True),
47
+ nn.MaxPool2d(kernel_size=2, stride=2),
48
+ nn.Conv2d(128, 128, kernel_size=3),
49
+ nn.ReLU(inplace=True),
50
+ nn.MaxPool2d(kernel_size=2, stride=2),
51
+ nn.Flatten()
52
+ )
53
+ self.fc = nn.Sequential(
54
+ nn.Linear(28672, 256),
55
+ nn.ReLU(inplace=True)
56
+ )
57
+
58
+ def forwardOne(self, x):
59
+ """
60
+ This method performs action of forward pass of single image.
61
+
62
+ It starts from extraction of features and then turning it to
63
+ vector of features, which is returned as an output.
64
+ """
65
+ x = self.featureExtractor(x)
66
+ x = self.fc(x)
67
+ return x
68
+
69
+ def forward(self, input1, input2):
70
+ """
71
+ This method performs forward pass for dataset sample which includes a pair of images.
72
+
73
+ It performs forward pass for both of the images at the same time and returns them as
74
+ outputs for further training.
75
+ """
76
+ output1 = self.forwardOne(input1)
77
+ output2 = self.forwardOne(input2)
78
+ return output1, output2
79
+
80
+ if __name__ == '__main__':
81
+ model = Siamese_nn().to('cuda')
82
+ summary(model, [(1, 153, 136), (1, 153, 136)], 1)
83
+ # %%