Spaces:
Sleeping
Sleeping
Sudipta Nayak
commited on
Commit
·
d7fee5d
1
Parent(s):
a995dbc
change-1
Browse files
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
|