Merge pull request yu4u#81 from moralfish/master

fix predicted genders label and add requirements.txt file for quick setup
zwl1671 · Feb 11, 2019 · 9410a57 · 9410a57
2 parents d2ab01f + d708e3b
commit 9410a57
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/demo.py b/demo.py
@@ -124,7 +124,7 @@ def main():
             # draw results
             for i, d in enumerate(detected):
                 label = "{}, {}".format(int(predicted_ages[i]),
-                                        "F" if predicted_genders[i][0] < 0.5 else "M")
+                                        "M" if predicted_genders[i][0] < 0.5 else "F")
                 draw_label(img, (d.left(), d.top()), label)
 
         cv2.imshow("result", img)

diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,11 @@
+opencv-python
+dlib
+imutils
+tensorflow
+Keras
+scipy
+numpy
+Pandas
+tqdm
+h5py
+tables