similarity code is updated08

app/Hackathon_setup/face_recognition.py

@@ -91,8 +91,9 @@ def get_similarity(img1, img2):
 
     # YOUR CODE HERE, return similarity measure using your model
     feature_net.eval()
-    normalized_face1 = F.normalize(face1, p=2, dim=1)
-    normalized_face2 = F.normalize(face2, p=2, dim=1)
+    print('face1.type', type(face1))
+    normalized_face1 = F.normalize(face1, p=2, dim=0)
+    normalized_face2 = F.normalize(face2, p=2, dim=0)
     output1,output2 = feature_net(normalized_face1,normalized_face2)
     euclidean_distance = F.pairwise_distance(output1, output2)
     #pairwise - more distance means less similarity