face expressions

app/Hackathon_setup/exp_recognition.py

@@ -23,6 +23,7 @@ import torch.nn.functional as F
 
 # Current_path stores absolute path of the file from where it runs.
 current_path = os.path.dirname(os.path.abspath(__file__))
+classes = {0: 'ANGER', 1: 'DISGUST', 2: 'FEAR', 3: 'HAPPINESS', 4: 'NEUTRAL', 5: 'SADNESS', 6: 'SURPRISE'}
 
 
 #1) The below function is used to detect faces in the given image.
@@ -171,4 +172,4 @@ def get_expression(img):
         probs = F.softmax(outputs, dim=1)
         predicted_class = probs.argmax(dim=1).item()
 
-    return predicted_class
+    return classes[predicted_class]