Sudipta Nayak commited on
Commit
d7fee5d
·
1 Parent(s): a995dbc
app/Hackathon_setup/face_recognition.py CHANGED
@@ -113,6 +113,7 @@ def get_face_class(img1):
113
  ##Better Hint: Siamese experiment is covered in one of the labs
114
 
115
  transform = transforms.Compose([
 
116
  transforms.Resize((100, 100)), # Resize the image to the desired size
117
  transforms.ToTensor(), # Convert the image to a PyTorch tensor
118
  transforms.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5]) # Normalize if needed
 
113
  ##Better Hint: Siamese experiment is covered in one of the labs
114
 
115
  transform = transforms.Compose([
116
+ transforms.Grayscale(num_output_channels = 1),
117
  transforms.Resize((100, 100)), # Resize the image to the desired size
118
  transforms.ToTensor(), # Convert the image to a PyTorch tensor
119
  transforms.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5]) # Normalize if needed