0xrushi commited on
Commit
cec0650
·
1 Parent(s): 05e29e2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +32 -1
README.md CHANGED
@@ -1,3 +1,34 @@
1
  ---
2
- license: apache-2.0
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ library_name: keras
3
  ---
4
+
5
+ ## Model description
6
+ This repo contains the model for the notebook [Image similarity estimation using a Siamese Network with a contrastive loss](https://keras.io/examples/vision/siamese_contrastive/).
7
+
8
+ Full credits go to Mehdi
9
+
10
+ Reproduced by [Rushi Chaudhari](https://github.com/rushic24)
11
+
12
+ [Siamese Networks](https://en.wikipedia.org/wiki/Siamese_neural_network) are neural networks which share weights between two or more sister networks, each producing embedding vectors of its respective inputs.
13
+
14
+ In supervised similarity learning, the networks are then trained to maximize the contrast (distance) between embeddings of inputs of different classes, while minimizing the distance between embeddings of similar classes, resulting in embedding spaces that reflect the class segmentation of the training inputs.
15
+
16
+ ## Dataset
17
+ [MNIST dataset](https://www.tensorflow.org/datasets/catalog/mnist) of handwritten digits
18
+
19
+ ## Training procedure
20
+
21
+ ### Training hyperparameters
22
+
23
+ The following hyperparameters were used during training:
24
+
25
+ ```
26
+ epochs = 10
27
+ batch_size = 16
28
+ margin = 1
29
+ ```
30
+
31
+ ### Training results
32
+
33
+ ![Contrastive loss](./contrastiveloss.png)
34
+ ![Accuracy](./accuracy.png)