jeevana commited on
Commit
225171a
1 Parent(s): 0609fff
app/Hackathon_setup/exp_recognition.py CHANGED
@@ -63,9 +63,8 @@ def get_expression(img):
63
  ##########################################################################################
64
 
65
  face = detected_face(img)
 
66
  print("type of face", type(face))
67
- print('face', face.size())
68
- face = img
69
  if face==0:
70
  face = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2GRAY))
71
  face = trnscm(face).unsqueeze(0)
 
63
  ##########################################################################################
64
 
65
  face = detected_face(img)
66
+ face = face.convert("RGB")
67
  print("type of face", type(face))
 
 
68
  if face==0:
69
  face = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2GRAY))
70
  face = trnscm(face).unsqueeze(0)