Spaces:
Running
Running
Pallavi Bhoj
commited on
Commit
·
405f037
1
Parent(s):
a54e4a9
Update face_recognition.py
Browse files
app/Hackathon_setup/exp_recognition_model.py
CHANGED
|
@@ -15,7 +15,7 @@ classes = {0: 'ANGER', 1: 'DISGUST', 2: 'FEAR', 3: 'HAPPINESS', 4: 'NEUTRAL', 5:
|
|
| 15 |
# Example Network
|
| 16 |
class facExpRec(torch.nn.Module):
|
| 17 |
def __init__(self):
|
| 18 |
-
super(facExpRec, self).
|
| 19 |
self.conv1 = nn.Conv2d(in_channels=1, out_channels=16, kernel_size=3)
|
| 20 |
|
| 21 |
self.conv2 = nn.Conv2d(in_channels=16, out_channels=64, kernel_size=3)
|
|
|
|
| 15 |
# Example Network
|
| 16 |
class facExpRec(torch.nn.Module):
|
| 17 |
def __init__(self):
|
| 18 |
+
super(facExpRec, self).__init__()
|
| 19 |
self.conv1 = nn.Conv2d(in_channels=1, out_channels=16, kernel_size=3)
|
| 20 |
|
| 21 |
self.conv2 = nn.Conv2d(in_channels=16, out_channels=64, kernel_size=3)
|