model hinzugefügt

projectmood/process_model.py

@@ -12,7 +12,7 @@ logging.basicConfig(level=logging.DEBUG, format='%(asctime)s %(levelname)-8s %(m
 args = sys.argv # liest Input Parameter
 logging.debug('Fisherface training initialized')
 
-file = open("gray.csv", "w")
+file = open("{}.csv".format('_'.join(args[1:]).lower()), "w")
 
 def _get_faces_from_emotion(emotion):
     """
@@ -34,8 +34,8 @@ def image_preprocessing(image):
     """
     img = cv2.imread(image)  # open image
     gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)  # convert to grayscale
-    equal = cv2.equalizeHist(gray)
-    return equal
+    blur = cv2.GaussianBlur(gray, (5, 5), 0)
+    return blur
 
 def make_sets():
     training_data = []