tsaddev commited on
Commit
416c572
·
1 Parent(s): d71985a

Update app/Hackathon_setup/face_recognition.py

Browse files
app/Hackathon_setup/face_recognition.py CHANGED
@@ -78,7 +78,7 @@ def get_similarity(img1, img2):
78
  siameseNetwork.load_state_dict(model['net_dict'])
79
 
80
  # YOUR CODE HERE, return similarity measure using your model
81
- out1, out2 = siameseNetwork(img1, img2)
82
  euclidean_distance = F.pairwise_distance(out1, out2)
83
 
84
  return euclidean_distance
 
78
  siameseNetwork.load_state_dict(model['net_dict'])
79
 
80
  # YOUR CODE HERE, return similarity measure using your model
81
+ out1, out2 = siameseNetwork(face1, face2)
82
  euclidean_distance = F.pairwise_distance(out1, out2)
83
 
84
  return euclidean_distance