sableenkaur27/ReefSentinel-Backend
0
1�
2 0@�i � � � U d Z ddlmZ ddlZddlZddlZddlmZ g d�g d�g d�g d�g d �d3�Z d e4d < d
ddddd5�Z de6d< dddddd�Z de7d< d d d �Z
d d!d �Zd"d �Zy)#u^ Five-class overlay: colored labels 1–5 only; class 0 keeps original pixels (notebook style).� )� annotationsN)�Image)� �� �� )r �� �d )� �< �P )� �U �# )� � r )� � � � � zdict[int, list[int]]�8ID2COLOR_5�9live_coral�bleached_coral�10dead_coral�algae�benthic_substratezdict[int, str]�CLASS_KEY_BY_IDz11Live CoralzBleached Coralz12Dead Coral�AlgaezBenthic Substrate)r r r r r zdict[str, str]�CLASS_LABEL_BY_KEYc �4 � t j | j d� t j �� }|j � }t dd� D ]G }||k( }t j t | t j �� }d|z13 || z ||z z ||< �I t j |dd� j t j � }t j � }t j |� j |dd� � |j � S )14N�RGB)�dtyper � r r �PNGT)�format�optimize)�np�array�convert�float32�copy�ranger �clip�astype�uint8�io�BytesIOr � fromarray�save�getvalue) �image�seg6�alpha�base�out�cid�m�col�bufs �KC:\Users\sable\OneDrive\Desktop\coral\backend\app\services\visualization.py� overlay_five_class_png_bytesr? % s� � �
�8�8�E�M�M�%�(��15�16� ;�D�17�)�)�+�C��Q��{� 4�� �C�K���h�h�z�#��b�j�j�9���e�)�s�1�v�%��� �3��A��4�
�'�'�#�q�#�18 �19%�20%�b�h�h�21/�C�22 �*�*�,�C� �O�O�C�� � �c�%�$� �?� �<�<�>�� c �` � t j t | ||� � j d� S )N�ascii)�base64�standard_b64encoder? �decode)r5 r6 r7 s r> �overlay_five_class_base64rF 2 s) � � � $� $�%A�%��u�%U� V� ]� ]�^e� f�fr@ c
�� � g }t dd� D ]X }t | }t | }|j |t | | dj23 |� t
| j |d� � d�� �Z |S )Nr r# z#{:02X}{:02X}{:02X}g )� class_key�label�rgb�hex�24percentage)r, r r �appendr r% �float�get)�benthic_percentages�legendr: �keyrJ s r> �#build_color_legend_with_percentagesrS 6 s| � ��F��Q��{� 25�� �c�"����o���
�
� �+�C�0��3�,�3�3�S�9� #�$7�$;�$;�C��$E� F�
� 26� 27� �Mr@ )g �?)r5 � Image.Imager6 �28np.ndarrayr7 rN �return�bytes)r5 rT r6 rU r7 rN rV �str)rP zdict[str, float]rV z29list[dict])�__doc__�30__future__r rC r0 �numpyr' �PILr r �__annotations__r r r? rF rS � r@ r> �<module>r_ s� �� d� "�
� � � � ����� $�31� � � ����� #�� � � �&� �
�,� &� �N� �32�g�r@