frozencherry/Forgery-Localization-App
1
1�
2 j}�h� � � � d dl Zd dlmZ G d� dej � Z G d� dej � Z G d� dej � Zy) � N)�layersc �, � � e Zd Zd� fd� Zd� Zd� Z� xZS )�ChannelAttentionc �: �� t t | �3 di |�� || _ y �N� )�superr �__init__�ratio)�selfr �kwargs� __class__s ��?/home/ubuntu/Documents/forgery-localization-app/custom_layer.pyr4 zChannelAttention.__init__ s �� �
� ��.�8��8���5� c � � |d }t j || j z ddd�� | _ t j |dd�� | _ y )N������relu� he_normalT)�6activation�kernel_initializer�use_bias)r r )r �Denser �shared_dense_one�shared_dense_two)r � input_shape�channelss r �buildzChannelAttention.build7 sJ � � �r�?�� &� � �X�t�z�z�-A�f�it� @D� !E�� � &� � �X�+�`d� e�� r c � � t j � |� } t j dd|j d f� |� }| j |� }| j |� } t j � |� } t j dd|j d f� |� }| j |� }| j |� }t j j ||z � } t j � ||g� S )N� ) r �GlobalAveragePooling2D�Reshape�shaper r �GlobalMaxPooling2D�tf�nn�sigmoid�Multiply)r �inputs�avg_pool�max_pool� cbam_features r �callzChannelAttention.call s� � �2�6�0�0�2�6�:��<�6�>�>�1�a�� � ��):�";�<�X�F���(�(��2���(�(��2��.�6�,�,�.�v�6��<�6�>�>�1�a�� � ��):�";�<�X�F���(�(��2���(�(��2���u�u�}�}�X��%8�9� � �v��� �&�,�!7�8�8r )� ��__name__�8__module__� __qualname__r9 r r, �
__classcell__�r s @r r r s �� ��f�10 119r r c �, � � e Zd Zd� fd� Zd� Zd� Z� xZS )�SpatialAttentionc �: �� t t | �12 di |�� || _ y r )r r5 r13 � kernel_size)r r7 r
r s �r r14 zSpatialAttention.__init__ s �� �
� ��.�8��8�&��r c �n � t j d| j | j fddd�� | _ y )Nr �samer F)�paddingr r )r �Conv2Dr7 �conv2d�r r s r r zSpatialAttention.build s5 � � �m�m�A��(8�(8�$�:J�:J�'K�U[�p{� GL� M�� r c �> � t j d� � |� } t j d� � |� } t j d�� ||g� }| j |� } t j d� |� } t j15 � ||g� S )Nc �2 � t j | dd�� S �N� T)�axis�keepdims)r$ � reduce_mean��xs r �<lambda>z'SpatialAttention.call.<locals>.<lambda>$ s � �2�>�>�!�!�d�+S� r c �2 � t j | dd�� S r@ )r$ �16reduce_maxrE s r rG z'SpatialAttention.call.<locals>.<lambda>% s � �2�=�=���T�+R� r rA )rB r& )r �Lambda� Concatenater< �17Activationr' )r r( r) r* �concatr+ s r r, zSpatialAttention.call# s� � �T�6�=�=�!S�T�U[�\��S�6�=�=�!R�S�TZ�[��+��#�#��+�X�x�,@�A���{�{�6�*� �3�v�(�(��3�L�A� � �v��� �&�,�!7�8�8r )� r. r3 s @r r5 r5 s �� �'�M�9r r5 c �* � � e Zd Z� fd�Zd� Zd� Z� xZS )�CBAMc �t �� t t | �18 di |�� d}d}t |� | _ t |� | _ y )Nr- rN r )r rP r19 r �channel_attentionr5 �spatial_attention)r r
r r7 r s �r r20 z
CBAM.__init__, s9 �� �
�d�D�"�,�V�,���� �!1�%�!8�� �!1�+�!>�� r c �J � | j |� }| j |� }|S �N)rR rS )r r( r+ s r r, z CBAM.call3 s( � ��-�-�f�5� ��-�-�l�;� ��r c � � y rU r r= s r r z21CBAM.build8 s � � r )r/ r0 r1 r22 r, r r2 r3 s @r rP rP + s �� �?� �23 r rP )�24tensorflowr$ �tensorflow.kerasr �Layerr r5 rP r r r �<module>rZ s; �� � #�9�v�|�|� 9�,9�v�|�|� 9� �6�<�<� r