Sorobot/Document-modify-Toolkit
0
1U
2 h�Hf � @ s@ d dl mZ ddlmZ G dd� dej�ZG dd� dej�ZdS )� N� )� weight_init_c s* e Zd ZdZd� fdd� Zdd� Z� ZS ) �
ResidualBlockz)Residual Block with custom normalization.�groupr c sX t t| ��� tj||dd|d�| _tj||ddd�| _t�� | _|dkr�|d }t� ||�| _3t� ||�| _ |dks�t� ||�| _ n�|dkr�t�
|�| _4t�
|�| _ |dks�t�
|�| _ nh|dkr�t�|�| _5t�|�| _ |dks�t�|�| _ n2|d k�r$t�� | _6t�� | _ |dk�s$t�� | _ |dk�r6d | _n t�tj||d|d7�| j �| _d S ) N� r )�padding�stride)r r � �batch�instance�none�r )�superr �__init__�nn�Conv2D�conv1�conv2�ReLU�relu� GroupNorm�norm1�norm2�norm3� BatchNorm2D�InstanceNorm2D�8Sequential�9downsample)�self� in_planes�planes�norm_fnr �10num_groups�� __class__� �*D:\CodeProjects\as1\Transform\extractor.pyr s> 11 121314151617 �zResidualBlock.__init__c C sR |}| � | �| �|���}| � | �| �|���}| jd k rD| �|�}| � || �S �N)r r r r r r )r �x�yr% r% r& �forward- s 1819zResidualBlock.forward)r r )�__name__�20__module__� __qualname__�__doc__r r* �
__classcell__r% r% r# r& r s $r c s4 e Zd ZdZd � fdd� Zd dd�Zd d21� Z� ZS )
� BasicEncoderz(Basic Encoder with custom normalization.� r22 c s2 t t| ��� || _| jdkr.t�dd�| _nD| jdkrFt�d�| _n,| jdkr^t�d�| _n| jdkrrt� � | _tj23dddd dd24�| _ t� � | _
d| _| jdd d �| _| jd
d d �| _| jdd d �| _t�25d|d �| _| �� D ]L}t|tj26��rt|jdddd� q�t|tjtjtjf�r�t|dd dd� q�d S )Nr r �@ r27 r r r � � )r r r r
r1 �� �kaiming_normal_�fan_outr )�mode� nonlinearity�Constantg )�valueZ28bias_value)r r0 r r! r r r r r r r r r �relu1r � _make_layer�layer1�layer2�layer3r � sublayers�29isinstancer �weight)r �30output_dimr! �mr# r% r& r ; s6 313233343536 �zBasicEncoder.__init__r c C s> t | j|| j|d�}t ||| jdd�}||f}|| _tj|� S )Nr
r )r r r! r r )r �dimr r> r? �layersr% r% r& r= [ s37 zBasicEncoder._make_layerc C sJ | � |�}| �|�}| �|�}| �|�}| �|�}| �|�}| �|�}|S r' )r r r< r> r? r@ r )r r( r% r% r& r* c s 38394041424344zBasicEncoder.forward)r1 r45 )r )r+ r, r- r. r r= r* r/ r% r% r# r& r0 8 s 46r0 )� paddle.nnr � weight_initr �Layerr r0 r% r% r% r& �<module> s 2