tsaddev commited on
Commit
798f555
·
1 Parent(s): f79ef04

Update app/Hackathon_setup/exp_recognition_model.py

Browse files
app/Hackathon_setup/exp_recognition_model.py CHANGED
@@ -61,4 +61,4 @@ def rgb2gray(image):
61
 
62
  # Sample Transformation function
63
  #YOUR CODE HERE for changing the Transformation values.
64
- trnscm = transforms.Compose([rgb2gray, transforms.Resize((48,48)), transforms.ToTensor(), transforms.Normalize((0.5), (0.5))])
 
61
 
62
  # Sample Transformation function
63
  #YOUR CODE HERE for changing the Transformation values.
64
+ trnscm = transforms.Compose([transforms.Resize((48,48)),transforms.Grayscale(), transforms.ToTensor(), transforms.Normalize((0.5), (0.5))])