Update face_recognition.py

app/Hackathon_setup/face_recognition.py

@@ -113,4 +113,16 @@ def get_face_class(img1):
     ##YOUR CODE HERE, return face class here
     ##Hint: you need a classifier finetuned for your classes, it takes o/p of siamese as i/p to it
     ##Better Hint: Siamese experiment is covered in one of the labs
-    return "YET TO BE CODED"
+    face1 = trnscm(det_img1).unsqueeze(0)
+
+    feature_net = SiameseNetwork().to(device)
+    model = torch.load(current_path + '/siamese_model.t7', map_location=device)  ##
+    feature_net.load_state_dict(model['net_dict'])
+    feature_net.eval()
+
+    output1, output2 = feature_net(face1.to(device), face1.to(device))
+    output1 = output1.detach().numpy()
+
+    clf_model = load(current_path + '/clf_model.joblib')
+    label = clf_model.predict(output1)
+    return label