holmeshoo/beans_sorting
0
1�
2 ���e" � � � d dl Z d dlZd dlZd dlmZ d dlZd dlZd dlZdZ dZ3d� Z d� Z d� Z
d� Zedk r�ej Zed Zed4z Zed z Z eee� � \ ZZ e
eeed � � Z ee� � d ZeD ]&Z ed
ed � � eded � � �'d ej ed � � k r ed z
ZdS dS dS )� N)�keras�V c �z � t j |t j � � }t j |t t5 f�� � }|dz }t
j |d�� � }| � |d�� � }t |� � i }t t |� � � � D ]}|d | ||| < �|S )N)�dsizeg �o@r )�axis)�verbose) �cv2�cvtColor�
COLOR_BGR2RGB�resize� IMG_WIDTH�6IMG_HEIGHT�np� expand_dims�predict�print�range�len)�model�label�img�p_result�result�is �K/mnt/b/4J/experiment/AI/assignment/main/src/app/beans_sorting/prediction.py�7predictImgr s� � �8
�,�s�C� -�9.�10.�C�11
�*�S��J� 7�128�138�148�C�15
��+�C�16 �.��1�17%�18%�19%�C��}�}�S�!�}�,�,�H� �(�O�O�O�
�F�
�3�u�:�:�
�
� *� *��#�A�;�q�>��u�Q�x��� �M� c �L � t j |� � }t | ||� � S )N)r �imreadr )r r �
img_file_namer s r � predictFiler! s# � �20
�*�]�21#�22#�C� �e�U�C� (� (�(r c �� �� �fd�t j �� � D � � }g }|D ]F}|� |t | |t j � �|� � � � g� � �G|S )Nc � �� g | ]A}t j � t j � �|� � � � �?|��BS � )�os�path�isfile�join)�.0�f� folder_paths �r �23<listcomp>z!predictFolder.<locals>.<listcomp>" s> �� �d�d�d�q�r�w�~�~�b�g�l�l�S^�`a�Fb�Fb�7c�7c�d��d�d�dr )r% �listdir�appendr! r& r( )r r r+ � filenamesr �files ` r �
predictFolderr1 sx �� �d�d�d�d�B�J�{� 3� 3�d�d�d�I�
�F�� Z� Z���
�
�t�[���r�w�|�|�K�QU�7V�7V� W� W�X�Y�Y�Y�Y� �Mr c �� � t j � | � � }t |d� � 5 }t j |� � }d d d � � n # 1 swxY w Y ||d fS )N�rr )r �models�24load_model�open�json�load)� model_dir� label_labelr r0 r s r �loadModelAndLabelr; * s� � � �L� #� #�I� .� .�E�
�k�3� � � �4�� �$���� � � � � � � � � � � ���� � � � � �%��.� � s �A�A�A�__main__� z.h5z.json� zFile:z Predictions:) r7 �numpyr �25tensorflow�tfr r �sysr% r r
r r! r1 r; �__name__�argv�args�26model_name� model_f_name� label_f_namer r r r �countr �argmaxr$ r r �<module>rK s �� � � � � � � � � � � � � � � � � � � 27�28�29�30� 31�32�33�34� � � � �
�35� � � � � � )� )� )� � � �!� !� !� �z��� �8�D��a��J� ��%�L� ��'�L�$�$�\�<�@�@�L�E�5� �]�5�%��a��
1�
1�F� �E�&�M�M�M�
�E�
� $� $��
��g�q��t� � � �
��n�a��d�#�#�#�#��I�B�I�a��d�O�O���
��36����# �� �r