Sudipta Nayak commited on
Commit
0c82c9c
·
1 Parent(s): 10912ee

name change

Browse files
app/Hackathon_setup/face_recognition.py CHANGED
@@ -115,7 +115,7 @@ def get_face_class(img1):
115
  feature_net = SiameseNetwork() #Example Network
116
  feature_net = feature_net.to(device)
117
  current_path = 'app/Hackathon_setup'
118
- model = torch.load(current_path + '/siamese_model.t7', map_location=device)
119
  feature_net.load_state_dict(model['net_dict'])
120
 
121
  output1 = feature_net.forward_once(det_img1.to(device))
 
115
  feature_net = SiameseNetwork() #Example Network
116
  feature_net = feature_net.to(device)
117
  current_path = 'app/Hackathon_setup'
118
+ model = torch.load(current_path + '/siamese_model_recog.t7', map_location=device)
119
  feature_net.load_state_dict(model['net_dict'])
120
 
121
  output1 = feature_net.forward_once(det_img1.to(device))