Initial Comit

app/Hackathon_setup/exp_recognition_model.py

@@ -20,7 +20,7 @@ class facExpRec(torch.nn.Module):
   def __init__(self, num_classes=7):
     super(facExpRec, self).__init__()
     self.conv_layers = nn.Sequential(
-      nn.Conv2d(in_channels=3, out_channels=32, kernel_size=3, stride=1, padding=1),
+      nn.Conv2d(in_channels=1, out_channels=32, kernel_size=3, stride=1, padding=1),
       nn.ReLU(),
       nn.MaxPool2d(kernel_size=2, stride=2),
       nn.Conv2d(in_channels=32, out_channels=64, kernel_size=3, stride=1, padding=1),
@@ -32,7 +32,7 @@ class facExpRec(torch.nn.Module):
     )
     self.fc_layers = nn.Sequential(
       nn.Flatten(),
-      nn.Linear(128 * 16 * 16, 512),
+      nn.Linear(128 * 6 * 6, 512),
       nn.ReLU(),
       nn.Dropout(0.5),
       nn.Linear(512, num_classes)