Recognition

app/Hackathon_setup/face_recognition_model.py

@@ -61,7 +61,21 @@ class Siamese(torch.nn.Module):
 ## classifier = nn.Sequential(nn.Linear(64, 64), nn.BatchNorm1d(64), nn.ReLU(), nn.Linear...)           ##
 ##########################################################################################################
 
-# YOUR CODE HERE for pytorch classifier
+# Load the existing trained DecisionTree classifier
+import joblib
+import os
+
+# Get the current directory path
+current_path = os.path.dirname(os.path.abspath(__file__))
+
+# Load the pre-trained DecisionTree classifier
+try:
+    classifier_path = os.path.join(current_path, 'decision_tree_model.sav')
+    classifier = joblib.load(classifier_path)
+    print(f"✓ Loaded DecisionTree classifier with {len(classifier.classes_)} classes")
+except Exception as e:
+    print(f"✗ Error loading DecisionTree classifier: {e}")
+    classifier = None
 
 # Definition of classes as dictionary
 classes = ['Person0','Person1','Person2','Person3','Person4','Person5','Person6']