adelgado/image_classifier
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add more debug

Browse source
This commit is contained in:
Antonio J. Delgado 2021-09-06 16:10:57 +03:00
parent 6e183640f8
commit 6e73345021
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
image_classifier/image_classifier.py
View file

@ -74,10 +74,12 @@ class image_classifier:
def load_known_people(self): def load_known_people(self):
known_people = list() known_people = list()
self._log.debug(f"Looking for known faces in directory '{self.faces_directory}'...")
if os.access(self.faces_directory, os.R_OK): if os.access(self.faces_directory, os.R_OK):
with os.scandir(self.faces_directory) as faces_items: with os.scandir(self.faces_directory) as faces_items:
for entry in faces_items: for entry in faces_items:
if not entry.name.startswith('.') and entry.is_file(): if not entry.name.startswith('.') and entry.is_file():
self._log.debug(f"Detecting people in known faces '{enrty.name}'...")
person = dict() person = dict()
person['filename'] = face_recognition.load_image_file(self.faces_directory + os.sep + entry.name) person['filename'] = face_recognition.load_image_file(self.faces_directory + os.sep + entry.name)
person['name'] = os.path.splitext(self.faces_directory + os.sep + entry.name)[0] person['name'] = os.path.splitext(self.faces_directory + os.sep + entry.name)[0]