Refactoring

projectmood/webcam.py

@@ -22,7 +22,7 @@ def _load_emoticons(emotions):
     return [nparray_as_image(cv2.imread('resources/graphics/%s.png' % emotion, -1), mode=None) for emotion in emotions]
 
 
-def show_webcam_and_run(model, emoticons, window_size=None, window_name='webcam', update_time=10):
+def show_webcam_and_run(model, emoticons, window_size=(800, 800), window_name='Mood Expression', update_time=1):
     """
     Shows webcam image, detects faces and its emotions in real time and draw emoticons over those faces.
     :param model: Learnt emotion detection model.
@@ -65,10 +65,8 @@ if __name__ == '__main__':
     emotions = ['neutral', 'anger', 'disgust', 'happy', 'surprise']
     emoticons = _load_emoticons(emotions)
 
-
     # load mode
     fisher_face = cv2.face.FisherFaceRecognizer_create()
     fisher_face.read('basis_data/models/detection_model.xml')
     # use learnt model
-    window_name = 'WEBCAM (press ESC to exit)'
-show_webcam_and_run(fisher_face, emoticons, window_size=(800, 800), window_name=window_name, update_time=1)
+show_webcam_and_run(fisher_face, emoticons)