a
�Lg� � @ s� d dl Z d dlZd dlmZ d dlmZmZ dZdd� ee�D �Z g g Z
Zee �D ]B\Z
Zee e
Ze �ee j�Ze
�ejeejd�� e�e
� qTejeejd�Ze j�� Ze�e�e
�e�e�� ed� dS ) � N)�listdir)�isfile�joinz./Myfaces/user/c C s g | ]}t tt|��r|�qS � )r r � data_path)�.0�fr r �9/Users/pranavzagade/PycharmProjects/LuciaAI/trainModel.py�
<listcomp> � r
)�dtypezModel trained sucessefully)�cv2�numpy�np�osr �os.pathr r r Z onlyfilesZ
Training_DataZLabels� enumerate�i�filesZ
image_path�imread�IMREAD_GRAYSCALEZimages�append�asarray�uint8�int32�face�LBPHFaceRecognizer_create�model�train�printr r r r �<module> s