Spaces:
Runtime error
Runtime error
Commit ·
b9ac5cf
1
Parent(s): 2a9d464
updated
Browse files
app/Hackathon_setup/exp_recognition.py
CHANGED
|
@@ -71,8 +71,7 @@ def get_expression(img):
|
|
| 71 |
face = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2GRAY))
|
| 72 |
|
| 73 |
with torch.no_grad():
|
| 74 |
-
|
| 75 |
-
tensor = transform(image)
|
| 76 |
output = face_det_net(face)
|
| 77 |
_, predicted = torch.max(output, 1)
|
| 78 |
|
|
|
|
| 71 |
face = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2GRAY))
|
| 72 |
|
| 73 |
with torch.no_grad():
|
| 74 |
+
face = trnscm(face)
|
|
|
|
| 75 |
output = face_det_net(face)
|
| 76 |
_, predicted = torch.max(output, 1)
|
| 77 |
|
app/Hackathon_setup/exp_recognition_model.py
CHANGED
|
@@ -51,4 +51,4 @@ def rgb2gray(image):
|
|
| 51 |
|
| 52 |
# Sample Transformation function
|
| 53 |
#YOUR CODE HERE for changing the Transformation values.
|
| 54 |
-
trnscm = transforms.Compose([rgb2gray, transforms.Resize((
|
|
|
|
| 51 |
|
| 52 |
# Sample Transformation function
|
| 53 |
#YOUR CODE HERE for changing the Transformation values.
|
| 54 |
+
trnscm = transforms.Compose([rgb2gray, transforms.Resize((100,100)), transforms.ToTensor()])
|