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

Update emotion_gender_processor.py

parent 9bcdeaae
No related branches found
No related tags found
No related merge requests found
......@@ -32,7 +32,7 @@ class EGProcessor:
# loading models
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.gender_classifier = load_model(models_dir + '/gender_mini_XCEPTION.01-0.94.hdf5', compile=False)
self.gender_classifier = load_model(models_dir + '/gender_models/simple_CNN.81-0.96.hdf5', compile=False)
if not os.path.exists(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