Update

app/app_utils.py

@@ -17,7 +17,6 @@ from app.face_utils import get_box, display_info
 from app.config import DICT_EMO, config_data
 from app.plot import statistics_plot
 
-
 mp_face_mesh = mp.solutions.face_mesh
 
 
@@ -91,7 +90,7 @@ def preprocess_video_and_predict(video):
                     startX, startY, endX, endY  = get_box(fl, w, h)
                     cur_face = frame_copy[startY:endY, startX: endX]
 
-                    if (count_frame-1)%5 == 0:
+                    if (count_frame-1)%config_data.FRAME_DOWNSAMPLING == 0:
                         cur_face_copy = pth_processing(Image.fromarray(cur_face)).to(config_data.DEVICE)
                         features = torch.nn.functional.relu(pth_model_static.extract_features(cur_face_copy)).cpu().detach().numpy()
         

config.toml

@@ -1,5 +1,6 @@
 APP_VERSION = "0.2.0"
 DEVICE = "cpu"
+FRAME_DOWNSAMPLING = 5
 
 [model_static]
 url = "https://huggingface.co/ElenaRyumina/face_emotion_recognition/resolve/main/FER_static_ResNet50_AffectNet.pth"