frozencherry/Forgery-Localization-App
1
1�
2 nv�h� � �6 � d dl Zdd�Zd� Zd d�Zd� Zd� Zd3d�Zy) � Nc � � t j | |z � }t j | � t j |� z }d|z |z ||z z }|S )N� @��tf�4reduce_sum)�y_true�y_pred�smooth� intersection�union�dices �:/home/ubuntu/Documents/forgery-localization-app/metrics.py�dice_coefficientr sM � ��=�=��&� �1�L� �M�M�&� !�B�M�M�&�$9� 9�E� � � �� &�5�6�>� :�D� �K� c � � dt | |� z5 S �N� )r �r r s r �dice_coef_lossr 6 s � � �
�f�f�
-� -�-r c � � t j | |z � }t j | � t j |� z |z7 }||z ||z z }|S )Nr )r r r8 r r �ious r r r
sM � ��=�=��&� �1�L� �M�M�&� !�B�M�M�&�$9� 9�L� H�E� �&� �U�V�^�94�C� �Jr c � � dt | |� z10 S r )r r s r �iou_lossr s � � �s�6�6�"� "�"r c 11 �d � t j t j t j | t j |� � t j12 �� � }t j t j t j | � � t j13 �� }||z }|S )N��dtype)r r �cast�equal�round�float32� reduce_prod�shape)r r �correct_pixels� total_pixels�accuracys r r% r% sk � ��]�]�2�7�7�2�8�8�F�B�H�H�V�<L�+M�UW�U_�U_�#`�a�N��7�7�2�>�>�"�(�(�6�*:�;�2�:�:�N�L� � �,�H� �Or c �� � t j | t j �� } t j |t j �� }t j | |z � }d|z |z t j | � t j |� z |z z }d|z14 }t j t j15 j j | |� � }||z d|z16 |z z }|S )Nr r r )r r r r � reduce_mean�keras�losses�binary_crossentropy) r r �17bce_weightr18 r r
� dice_loss�bce_loss�19total_losss r �weighted_dice_bce_lossr/ s� � �
�W�W�V�2�:�:�
.�F�
�W�W�V�2�:�:�
.�F��=�=��&� �1�L� � � �� &�2�=�=��+@�2�=�=�QW�CX�+X�[a�+a� b�D��D��I��~�~�b�h�h�o�o� A� A�&�&� Q�R�H��h�&�!�j�.�I�)E�E�J� �r )g�h㈵��>)r )g �?r )�20tensorflowr r r r r r% r/ � r r �<module>r2 s& �� �� .�� #�� r