add02

app/Hackathon_setup/exp_recognition.py

@@ -63,10 +63,11 @@ def get_expression(img):
     ##########################################################################################
 
     face = detected_face(img)
+    print("type of face", type(face))
     print('face', face.size())
-    #face = img
-    # if face==0:
-    #     face = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2GRAY))
+    face = img
+    if face==0:
+        face = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2GRAY))
     face = trnscm(face).unsqueeze(0)
     # YOUR CODE HERE, return expression using your model
     file = open(current_path + "/exp_model.t7")