Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -50,12 +50,12 @@ class SiameseNetwork(nn.Module):
|
|
| 50 |
x = self.sigmoid(x)
|
| 51 |
return x
|
| 52 |
|
| 53 |
-
def preprocess_image_siamese(
|
| 54 |
transform = transforms.Compose([
|
| 55 |
transforms.Resize((224, 224)),
|
| 56 |
transforms.ToTensor()
|
| 57 |
])
|
| 58 |
-
img = Image.open(
|
| 59 |
return transform(img)
|
| 60 |
|
| 61 |
def preprocess_image_svm(img):
|
|
@@ -95,10 +95,8 @@ def verify(image, model, person):
|
|
| 95 |
siamese = SiameseNetwork()
|
| 96 |
siamese.load_state_dict(torch.load(f'siamese_{person.lower()}.pth'))
|
| 97 |
siamese.eval()
|
| 98 |
-
|
| 99 |
-
face = Image.open(temp_face_path)
|
| 100 |
|
| 101 |
-
face = preprocess_image_siamese(
|
| 102 |
|
| 103 |
with torch.no_grad():
|
| 104 |
output = model(face)
|
|
|
|
| 50 |
x = self.sigmoid(x)
|
| 51 |
return x
|
| 52 |
|
| 53 |
+
def preprocess_image_siamese(temp_face_path):
|
| 54 |
transform = transforms.Compose([
|
| 55 |
transforms.Resize((224, 224)),
|
| 56 |
transforms.ToTensor()
|
| 57 |
])
|
| 58 |
+
img = Image.open(temp_face_path).convert("RGB")
|
| 59 |
return transform(img)
|
| 60 |
|
| 61 |
def preprocess_image_svm(img):
|
|
|
|
| 95 |
siamese = SiameseNetwork()
|
| 96 |
siamese.load_state_dict(torch.load(f'siamese_{person.lower()}.pth'))
|
| 97 |
siamese.eval()
|
|
|
|
|
|
|
| 98 |
|
| 99 |
+
face = preprocess_image_siamese(temp_face_path)
|
| 100 |
|
| 101 |
with torch.no_grad():
|
| 102 |
output = model(face)
|