face recognition

app/Hackathon_setup/face_recognition_model.py

@@ -8,7 +8,14 @@ from torchvision import transforms
 
 # Sample Transformation function
 # YOUR CODE HERE for changing the Transformation values.
-trnscm = transforms.Compose([transforms.Resize((100,100)), transforms.ToTensor()])
+# trnscm = transforms.Compose([transforms.Resize((100,100)), transforms.ToTensor()])
+
+trnscm = transforms.Compose([
+    transforms.Grayscale(num_output_channels=1),
+    transforms.Resize((100, 100)),
+    transforms.ToTensor(),
+    transforms.Normalize(mean=[0.5], std=[0.5])
+])
 
 ##Example Network
 class Siamese(torch.nn.Module):