holmeshoo/beans_sorting
0
1�
2 ���e�
� �$ � d dl Zd dlZd dlZd dlZd dlZd dl Z3d dl Z g Z e �
ej dd� � � � e �
ej dd� � � � e �
ej dd� � � � ed� � g d �Zd4� Zd � Zd � Z ej ej � d
�� � �� � 5 Z ej d� � ej d� � 5 ej eded �� � Z ej d�� � Z ej d d�� � Z e� ee �� � d� � � Z! ej" d� � Z# ej$ � � 5 ej d� � Z% ej& � � Z'ddd� � n # 1 swxY w Y e#�( ee e ge%e'g� � � ddd� � n # 1 swxY w Y ej d � � 5 ej eded �� � Z ej d�� � Z ej d d�� � Z e� ee �� � d � � � Z! ej" d� � Z# ej$ � � 5 ej d� � Z% ej& d� � Z'ddd� � n # 1 swxY w Y ej d �� � Z)e#�( ee e ge)e'e%g� � � ddd� � n # 1 swxY w Y ddd� � n # 1 swxY w Y e*d k re�+ � � dS dS )!� Nz./model/white_beans.h5z./model/white_beans.jsonz./model/red_beans.h5z./model/red_beans.jsonz./model/black_beans.h5z./model/black_beans.jsonz model loaded)�white�red�blackc �8 � || v r| � |� � S dS )N�����)�index)�l�xs �D/mnt/b/4J/experiment/AI/assignment/main/src/app/beans_sorting/app.py�my_findr s � ��A�v�v��w�w�q�z�z���r� c �\ � t | \ }}t j | � � }t j |� � }t j ||||�� � }t j |||� � }t j |t j � � }t j5 |� � }i }|D ]}d||< �|D ]:} d}6d} | � � � D ]\ } }
| |
k r|
} | }7�||8xx dz
cc< �;t j
|� � }|� � � D ]"\ }}||xx t |� � z cc< �#d}d}|� � � D ]A\ }}|t |� � � � � dz z
}|t |� � dz z
}�B|dz dt |� � z z dz |z dz }|||fS ) N)�model�label�img�pointr � � �|�9�|:-:�|10)� model_label�cv2�imread�mprd�getArea� multiPredict�printResultImg�cvtColor�
COLOR_BGR2RGB�np�array�items�copy�deepcopy�len�str�upper)� file_name�nr r r r �result�sum�key�i� max_label�max_val�n_label�n_val� sum_for_txtr �v�keys�values�k�11table_texts r �multiPredeictionr: s� � � ��^�N�U�E�12
�*�Y�13 �14 �C� �L�� � �E�
�
�U�%�S��
N�
N�
N�F�15�16 �c�5�&�171�181�C�19
�,�s�C� -�20.�21.�C�22 �(�3�-�-�C�23 �C�� � ����C���
� � ��� ��� �g�g�i�i� $� $�N�G�U��%��� �� #� �� �I����!�������-��$�$�K�� � � � � � ���1� �A����#�f�+�+� ����� �D�
�F��!�!�#�#� � ���1� ��A��� � ����$�$���#�a�&�&�3�,� ������v��E�24�25� 2�2�U�:�V�C�d�J�J� ��Z� � r
c �r � t | \ }}t j ||| � � }d}d}|� � � D ]B\ }}|t |� � � � � dz z
}|t
|d� � dz z
}�C|dz dt |� � z z dz |z dz } t j | � � |fS )Nr z.2fr r r )26r �prd� predictFiler$ r( r) �formatr' �gr�Markdown)27r* r+ r r r, r6 r7 r8 r5 r9 s28 r r= r= = s� � � ��^�N�U�E�
�_�U�E�9�
5�
5�F� �D�
�F�� � ��� )� )���1� ��A��� � ����$�$���&��E�"�"�S�(�(������v��E�29�30� 2�2�U�:�V�C�d�J�J� �K�31� #� #�V� ,�,r
�orange)� primary_hue)�themez# Beans Sortingzsingle detectionz
beans variety)r �value�filepath)�typeF)rD �visible)�inputs�outputsc �R � t j t t | � � �� � S �N)rD �r? �Numberr �
beans_variety��choices r � changeModelrQ U � � ��9�7�=�&�#A�#A�B�B�B� Br
�Runr )�fnrH rI zMultiple detectionc �R � t j t t | � � �� � S rK rL rO s r rQ rQ h rR r
�numpy�__main__),�gradior? �32predictionr< �multiple_prdictionr �pandas�pdr rV r" r% r �append�loadModelAndLabel�printrN r r: r= �Blocks�themes�Soft�demor@ �Tab�Dropdown�dropdown�Image�picturerM � model_number�changerQ �Button�button�Row�text�Labelr �click�output�__name__�launch� r
r �<module>ru sD �� � � � � � � � � !� !� !� !� � � � � 33�34�35�36� � � � � � � � �� � � � �(�3�(�)A�C]�^�^� _� _� _� � � �(�3�(�)?�AY�Z�Z� [� [� [� � � �(�3�(�)A�C]�^�^� _� _� _� ��n� � � �� � �
�� � � � � �@ -� -� -� �R�Y�R�Y�^�^��^�9�9�:�:�:� %i�d��B�K�!�"�"�"� ���"� #� #� \� \� �2�;�}�O�=�YZ�K[�\�\�\���"�(�37�+�+�+�� �r�y�q�%�8�8�8� � � � ��,� � ?� ?� C� C� 38@� ?� C� ���5�!�!��
�R�V�X�X� � � �2�;�r�?�?�D� �B�H�J�J�E� � � � � � � � � � � ���� � � � � � � � �W�l�,C�d�TY�]� �[�[�[�#\� \� \� \� \� \� \� \� \� \� \���� \� \� \� \�$ 39���$� %� %� i� i� �2�;�}�O�=�YZ�K[�\�\�\���"�(�40�+�+�+�� !�r�y�q�%�8�8�8� � � � ��,� � ?� ?� C� C� 41@� ?� C����5�!�!��
�R�V�X�X� !� !� �2�;�r�?�?�D� �B�H�R�L�L�E� !� !� !� !� !� !� !� !� !� !� !���� !� !� !� !� ���w�'�'�'��� � �(�'�<�1H�SY�[`�bf�Rg� �h�h�h�#i� i� i� i� i� i� i� i� i� i� i���� i� i� i� i�)%i� %i� %i� %i� %i� %i� %i� %i� %i� %i� %i���� %i� %i� %i� %i�P �z����K�K�M�M�M�M�M� �s� �
!K)�.A<G �* F�42 G �F �G � F � G �= K)� G
�
K)�G
�K)�$A<K� !J
� K�
J �K�J �1K� K)�K �K)�K �K)�)K-�0K-