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)
|