Spaces:
Sleeping
Sleeping
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 + '/
|
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))
|