Mery3391/YAMNET
0
1�
2 D0�i^ � �� � S SK rS SKrS SKr\R " S5 rS \l \\R S' S SKr S SK3r4S SK r S SK
rS SKrS SKJrJr \" S5 \R( " S5 r\" S5 \ R, R$ R/ S5 r\" S 5 \" S5S 5 r\R( " \5 rSSS5 S S
SSS.r S\ S\ 4S jr g! , ( d f N = f)� N�
pkg_resourcesc �J � [ S U R S5 S S 5 5 $ )Nc 3 �8 # � U H n[ U5 v � M g 7f)N)�int)�.0�xs �+C:\Users\hp\Downloads\mon_projet\predict.py� <genexpr>�<lambda>.<locals>.<genexpr> s � � �-O�>N��c�!�f�f�>N�s ��.� )�tuple�split)�vs r �<lambda>r s � ��-O�a�g�g�c�l�2�A�>N�-O�(O� )�layers�modelszChargement de YAMNet...z!https://tfhub.dev/google/yamnet/1u Chargement du modèle emotion...zmodel_yamnet_darija.keraszChargement du scaler...zscaler_yamnet.pkl�rb�Angry�Happy�Neutral�Sad)r � � r
�6audio_path�returnc �� � [ R " U SS9u pUR [ R 5 n[ U5 u p#n[ R " USS9R 5 R SS5 n[ R U5 n[ R USS9n[ [ R " U5 5 n[! [ R" " U5 5 n[$ U [' US- S 5 [' [! US S 5 S- S 5 [' [! US S 5 S- S 5 [' [! US S 5 S- S 5 [' [! US S7 5 S- S 5 S .S .$ )
Ni�> )�srr )�axisr �����)�verbose�d r r
)r r r r )�emotion�8confidence�scores)�librosa�load�astype�np�float32� yamnet_model�tf� reduce_mean�numpy�reshape�scaler� transform�
emotion_model�predictr �argmax�float�max�LABELS�round) r �audio�_�9embeddings�mean_embedding�embedding_scaled� predictions�predicted_indexr% s r �predict_emotionrA sB � ��|�|�J�5�1�H�E� �L�L���� $�E�#�E�*��A�1��^�^�J�Q�7�=�=�?�G�G��2�N�N� �'�'��7�� �'�'�(8�!�'�D�K��"�)�)�K�0�1�O��r�v�v�k�*�+�J��/�*��J�� ,�a�0� �U�;�q�>�!�#4� 5�� ;�Q�?� �U�;�q�>�!�#4� 5�� ;�Q�?� �U�;�q�>�!�#4� 5�� ;�Q�?� �U�;�q�>�!�#4� 5�� ;�Q�?� 10� � r ) �importlib.metadata� importlib�types�sys�11ModuleTyper �
parse_version�modulesr/ r* r' �12tensorflowr- �tensorflow_hub�hub�pickle�tensorflow.kerasr r �printr( r, �keras�13load_modelr3 �open�fr1 r8 �str�dictrA � r r �<module>rV s� �� � � 14�� � � �1�
� O�
� � ,�� � �O� � � � � �
� +� � � ��x�x�;�<� � �(� )������*�*�+F�G�
� � � � �15 �t�$��
�[�[��^�F� %� ��Y�5� 9���� �� � %�$�s �*C�16C$