Tejaswini2403 commited on
Commit
b9ac5cf
·
1 Parent(s): 2a9d464
app/Hackathon_setup/exp_recognition.py CHANGED
@@ -71,8 +71,7 @@ def get_expression(img):
71
  face = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2GRAY))
72
 
73
  with torch.no_grad():
74
- transform = transforms.ToTensor()
75
- tensor = transform(image)
76
  output = face_det_net(face)
77
  _, predicted = torch.max(output, 1)
78
 
 
71
  face = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2GRAY))
72
 
73
  with torch.no_grad():
74
+ face = trnscm(face)
 
75
  output = face_det_net(face)
76
  _, predicted = torch.max(output, 1)
77
 
app/Hackathon_setup/exp_recognition_model.py CHANGED
@@ -51,4 +51,4 @@ def rgb2gray(image):
51
 
52
  # Sample Transformation function
53
  #YOUR CODE HERE for changing the Transformation values.
54
- trnscm = transforms.Compose([rgb2gray, transforms.Resize((48,48)), transforms.ToTensor()])
 
51
 
52
  # Sample Transformation function
53
  #YOUR CODE HERE for changing the Transformation values.
54
+ trnscm = transforms.Compose([rgb2gray, transforms.Resize((100,100)), transforms.ToTensor()])