classifier01

app/Hackathon_setup/face_recognition.py

@@ -118,13 +118,13 @@ def get_similarity(img1, img2):
 ##Caution: Don't change the definition or function name; for loading the model use the current_path for path example is given in comments to the function
 def get_face_class(img1):
     device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
-
+    classes = ['person1','person2','person6','person7']
     det_img1 = detected_face(img1)
     if det_img1 == 0:
         det_img1 = Image.fromarray(cv2.cvtColor(img1, cv2.COLOR_BGR2GRAY))
     img1 = trnscm(det_img1).unsqueeze(0)
     feature_net = Siamese()  #                                           ##
-    feature_classifier = MLPClassifier()
+    feature_classifier = MLPClassifier(input_size=5, hidden_size=2048, num_classes=4)
     model = torch.load(current_path + "/siamese_model.t7", map_location="cpu")  ##
     feature_net.load_state_dict(model["net_dict"])  ##
     #classifier