tsaddev commited on
Commit
3bd78ba
·
1 Parent(s): 1e160ee

Update app/Hackathon_setup/exp_recognition.py

Browse files
app/Hackathon_setup/exp_recognition.py CHANGED
@@ -70,6 +70,7 @@ def get_expression(img):
70
  face_det_net = facExpRec()
71
  model = torch.load(current_path + '/exp_recognition_net.t7', map_location=device)
72
  face_det_net.load_state_dict(model['net_dict'])
 
73
  face = trnscm(face).unsqueeze(0)
74
  print(face.shape)
75
  output = face_det_net(face)
 
70
  face_det_net = facExpRec()
71
  model = torch.load(current_path + '/exp_recognition_net.t7', map_location=device)
72
  face_det_net.load_state_dict(model['net_dict'])
73
+ face_det_net.eval()
74
  face = trnscm(face).unsqueeze(0)
75
  print(face.shape)
76
  output = face_det_net(face)