Added source code for te 4th ABAW competition

av-savchenko · Jul 21, 2022 · 24c7b22 · 24c7b22
1 parent 21822ef
commit 24c7b22
Show file tree

Hide file tree

Showing 5 changed files with 7,148 additions and 94 deletions.
diff --git a/models/affectnet_emotions/enet_b0_8_va_mtl.pt b/models/affectnet_emotions/enet_b0_8_va_mtl.pt
diff --git a/src/ABAW/abaw3_train.ipynb b/src/ABAW/abaw3_train.ipynb
@@ -768,7 +768,7 @@
     "model_name='mymobilenet_7_ft_sgd_model'\n",
     "num_classes=7\n",
     "\n",
-    "base_model=load_model('../models/affectnet_emotions/'+model_name+'.h5')\n",
+    "base_model=load_model('../../models/affectnet_emotions/'+model_name+'.h5')\n",
     "_,w,h,_=base_model.input.shape\n",
     "print(w,h)"
    ]
@@ -997,7 +997,7 @@
     ")\n",
     "\n",
     "print(PATH)\n",
-    "feature_extractor_model = torch.load('../models/affectnet_emotions/'+PATH)"
+    "feature_extractor_model = torch.load('../../models/affectnet_emotions/'+PATH)"
    ]
   },
   {