Skip to content
Snippets Groups Projects
Commit b35cef39 authored by Vesa Halenius's avatar Vesa Halenius
Browse files

Update emotion_gender_processor.py

parent 0ecebbf9
No related branches found
No related tags found
No related merge requests found
...@@ -32,7 +32,7 @@ class EGProcessor: ...@@ -32,7 +32,7 @@ class EGProcessor:
# loading models # loading models
self.face_detection = load_detection_model(models_dir + '/detection_models/haarcascade_frontalface_default.xml') self.face_detection = load_detection_model(models_dir + '/detection_models/haarcascade_frontalface_default.xml')
self.emotion_classifier = load_model(models_dir + '/emotion_models/fer2013_mini_XCEPTION.102-0.66.hdf5', compile=False) self.emotion_classifier = load_model(models_dir + '/emotion_models/fer2013_mini_XCEPTION.102-0.66.hdf5', compile=False)
self.gender_classifier = load_model(models_dir + '/gender_models/gender_mini_XCEPTION.00-0.90.hdf5', compile=False) self.gender_classifier = load_model(models_dir + '/gender_models/gender_mini_XCEPTION.01-0.94.hdf5', compile=False)
if not os.path.exists(dirname): if not os.path.exists(dirname):
os.mkdir(dirname) os.mkdir(dirname)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment