hololens/stable-diffusion-webui-depthmap-script
1
1o
2 ��f( � @ s� d dl mZ d dl mZ g d�Zdddddd�Zd d3d �ZG d d
� d
ej�ZG dd� dej�ZG dd� dej�Z d dd�Z4d dd�Z d dd�Z d dd�Z
d dd �ZdS ) � N)�ResNet�resnet18�resnet34�resnet50� resnet101� resnet152z9https://download.pytorch.org/models/resnet18-5c106cde.pthz9https://download.pytorch.org/models/resnet34-333f7ec4.pthz9https://download.pytorch.org/models/resnet50-19c8e357.pthz:https://download.pytorch.org/models/resnet101-5d3b4d8f.pthz:https://download.pytorch.org/models/resnet152-b121ed2d.pth)r r r r r � c C s t j| |d|ddd�S )z 3x3 convolution with padding� r F�� kernel_size�stride�padding�bias)�nn�Conv2d)� in_planes�5out_planesr � r �BF:\3dpicmaker\stable-diffusion-webui-depthmap-script\lib\Resnet.py�conv3x3 s �r c s* e Zd ZdZd� fdd� Zdd� Z� ZS )�6BasicBlockr Nc s^ t t| ��� t|||�| _t�|�| _tj dd�| _7t||�| _ t�|�| _ || _
|| _d S )NT��inplace)�superr �__init__r �conv1�NN� BatchNorm2d�bn1r �ReLU�relu�conv2�bn2�8downsampler ��self�inplanes�planesr r# �� __class__r r r s 9zBasicBlock.__init__c C s` |}| � |�}| �|�}| �|�}| �|�}| �|�}| jd ur%| �|�}||7 }| �|�}|S �N)r r r r! r" r# �r% �x�residual�outr r r �forward$ s 1011121314151617zBasicBlock.forward�r N��__name__�18__module__� __qualname__� expansionr r/ �
__classcell__r r r( r r s 19r c s* e Zd ZdZd� fdd� Zdd� Z� ZS ) �20Bottleneck� r Nc s� t t| ��� tj||ddd�| _t�|�| _tj||d|ddd�| _ t�|�| _21tj||| j ddd�| _ t�|| j �| _
tjdd�| _|| _|| _d S )Nr F)r r r r22 Tr )r r7 r r r r r r r r! r" r5 �conv3�bn3r r r# r r$ r( r r r : s � 23zBottleneck.__init__c C s~ |}| � |�}| �|�}| �|�}| �|�}| �|�}| �|�}| �|�}| �|�}| jd ur4| �|�}||7 }| �|�}|S r* )r r r r! r" r9 r: r# r+ r r r r/ G s 2425262728293031323334zBottleneck.forwardr0 r1 r r r( r r7 7 s
r7 c s0 e Zd Zd � fdd� Zd35dd�Zdd� Z� ZS ) r �� c s d| _ tt| ��� tjddddddd�| _t�d�| _ tj36dd�| _ tj ddd d37�| _
| �|d|d �| _| j|d |d dd
�| _| j|d|d dd
�| _| j|d|d dd
�| _| �� D ])}t|tj�rqtjj|jddd� q^t|tj�r�tj�|jd � tj�|jd � q^d S )N�@ r � � Fr38 Tr r )r r r
r � )r � i �fan_outr )�mode� nonlinearity)r& r r r r r r r r r r r � MaxPool2d�maxpool� _make_layer�layer1�layer2�layer3�layer4�modules�39isinstance�init�kaiming_normal_�weight� constant_r )r% �block�layers� num_classes�mr( r r r ` s( � ��zResNet.__init__r c C s� d }|dks| j ||j kr&t�tj| j ||j d|dd�t�||j ��}g }|�|| j |||�� ||j | _ td|�D ] }|�|| j |�� q>tj|� S )Nr F)r r r ) r& r5 r �40Sequentialr r r �append�range)r% rQ r' �blocksr r# rR �ir r r rF v s �� 41zResNet._make_layerc C s� g }| � |�}| �|�}| �|�}| �|�}| �|�}|�|� | �|�}|�|� | �|�}|�|� | �|�}|�|� |S r* ) r r r rE rG rV rH rI rJ )r% r, �featuresr r r r/ � s 424344454647484950515253zResNet.forward)r; �r )r2 r3 r4 r rF r/ r6 r r r( r r ^ s 54r Tc K � t tg d�fi |��}|S )zxConstructs a ResNet-18 model.55 Args:56 pretrained (bool): If True, returns a model pre-trained on ImageNet57 )r> r> r> r> �r r ��58pretrained�kwargs�modelr r r r � � r c K r\ )zxConstructs a ResNet-34 model.59 Args:60 pretrained (bool): If True, returns a model pre-trained on ImageNet61 �r r8 � r r] r^ r r r r � rb r c K r\ )zxConstructs a ResNet-50 model.62 Args:63 pretrained (bool): If True, returns a model pre-trained on ImageNet64 rc �r r7 r^ r r r r � � r c K r\ )zyConstructs a ResNet-101 model.65 Args:66 pretrained (bool): If True, returns a model pre-trained on ImageNet67 )r r8 � r re r^ r r r r � rf r c K r\ )zyConstructs a ResNet-152 model.68 Args:69 pretrained (bool): If True, returns a model pre-trained on ImageNet70 )r � �$ r re r^ r r r r � rb r r[ )T)�torch.nnr r �__all__�71model_urlsr �Moduler r7 r r r r r r r r r r �<module> s$ �72 '73=74 75 76 7778