CLYang617/RemoteSensingChangeDetection-RSCD.HA2F
0
1a
2 �o�gF � @ sZ d dl Z d dlmZ d dlmZ d dlm mZ G dd� dej�Z G dd� dej�Z3dS )� N)�modelsc s: e Zd Zd4� fdd� Zdd� Zedd� �Zdd � Z� ZS ) �SimAM�-C ��6?c s" t t| ��� t�� | _|| _d S )N)�superr �__init__�nn�Sigmoid� activaton�e_lambda)�selfr5 �� __class__� �9C:\Users\82607\Desktop\code\ChangeViT-master\model\dem.pyr s 6zSimAM.__init__c C s | j jd }|d| j 7 }|S )N�(z7lambda=%f))r
�__name__r8 )r �sr r r �__repr__
s zSimAM.__repr__c C s dS )NZsimamr r r r r �get_module_name s zSimAM.get_module_namec 9 C sn |� � \}}}}|| d }||jddgdd� �d�}|d|jddgdd�| | j d }| �|�} | S )N� � � T)�dim�keepdim� g �?)�size�mean�pow�sumr10 r )11r �x�b�c�h�w�nZx_minus_mu_square�yZ12att_weightr r r �forward s (13z
SimAM.forward)r ) r �14__module__� __qualname__r r � staticmethodr r& �
__classcell__r r r r r s15 16r c s$ e Zd Z� fdd�Zdd� Z� ZS )�17ThreeD_DEMc sR t t| ��� t�t�|d |ddd�t�|�tjdd��| _t � | _18t � | _ d S )Nr r r T)�inplace) r r+ r r �19Sequential�Conv2d� BatchNorm2d�ReLU�convr �att1�att2)r � in_channelsr r r r s 20�zThreeD_DEM.__init__c 21 C sv t �|| �}| j�|�}|| }|| | }|| | }t j||gdd�}| �|�}| j�|�}|| }|| } | S )Nr )r )�torch�absr2 r& �catr1 r3 )22r �x1�x2�diffZdiff_attZcon1Zcon2r7 Zcat_attZ fuse_featurer r r r& ' s 23 zThreeD_DEM.forward)r r' r( r r& r* r r r r r+ s r+ ) r5 �torch.nnr � torchvisionr �torch.nn.functional�24functional�F�Moduler r+ r r r r �<module> s25