Upload 4 files
Browse files
app/Hackathon_setup/exp_recognition.py
CHANGED
|
@@ -1,3 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
import numpy(__file__))import numpy as np
|
| 2 |
|
| 3 |
_MODEL = None
|
|
@@ -71,10 +78,4 @@ def get_expression(img):
|
|
| 71 |
|
| 72 |
return classes.get(pred, "UNKNOWN").capitalize()
|
| 73 |
|
| 74 |
-
import cv2
|
| 75 |
-
import torch
|
| 76 |
-
from PIL import Image
|
| 77 |
-
import os
|
| 78 |
-
|
| 79 |
-
from .exp_recognition_model import trnscm, load_model, classes
|
| 80 |
|
|
|
|
| 1 |
+
import cv2
|
| 2 |
+
import torch
|
| 3 |
+
from PIL import Image
|
| 4 |
+
import os
|
| 5 |
+
|
| 6 |
+
from .exp_recognition_model import trnscm, load_model, classes
|
| 7 |
+
|
| 8 |
import numpy(__file__))import numpy as np
|
| 9 |
|
| 10 |
_MODEL = None
|
|
|
|
| 78 |
|
| 79 |
return classes.get(pred, "UNKNOWN").capitalize()
|
| 80 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 81 |
|
app/Hackathon_setup/face_recognition.py
CHANGED
|
@@ -1,4 +1,3 @@
|
|
| 1 |
-
%%writefile face_recognition_model.py
|
| 2 |
import torch
|
| 3 |
import torch.nn as nn
|
| 4 |
from torchvision import transforms
|
|
|
|
|
|
|
| 1 |
import torch
|
| 2 |
import torch.nn as nn
|
| 3 |
from torchvision import transforms
|