dotslash21 / faceaccessserver Goto Github PK

Describe the bug
On line 47 of face_id.py,
https://github.com/dotslash21/realtime_face_identification/blob/ac0f50dee6acdb97232fdd5b46d647d5c0f06277/face_id.py#L45-L48

Current implementation to load the classification labels for the SVM uses os.listdir() function which gets all the file and folder names from the directory './aligned_img_db'. The issue is other than the directories there are several bounding_box_XXXXX.txt files which are also getting added to self.faceIds which wrongfully allows any unauthorized person to access.

To Reproduce
Steps to reproduce the behavior:

1. Follow the instructions here
2. Open the dashboard. (localhost:8080)
3. Click on VIEW FEED
4. See yourself authenticated and labeled bounding_box_XXXXX.txt

Expected behavior
self.faceIds should only have the immediate subdirectories name list as labels.