CoolFace
Apppublic

hololens/stable-diffusion-webui-depthmap-script

sourceHugging Faceupdated 2y agoView on Hugging Face
1likes
Resnet.cpython-311.pyc48 linesDownload Raw Back to __pycache__
12��f(����ddlmZddlmZgd�Zdddddd�Zdd3�ZGd�dej��ZGd
�dej��ZGd�dej��Z	dd�Z4dd�Zdd�Zdd�Z
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)rrrrr�c�6�tj||d|dd���S)z3x3 convolution with padding�r	F��kernel_size�stride�padding�bias)�nn�Conv2d)�	in_planes�5out_planesrs   �BF:\3dpicmaker\stable-diffusion-webui-depthmap-script\lib\Resnet.py�conv3x3rs(��
�9�Y�6��&��U�,�,�,�,�c�*��eZdZdZd�fd�	Zd�Z�xZS)�7BasicBlockr	Nc�`��tt|�����t|||��|_tj|��|_tj	d���|_8t||��|_tj|��|_||_
||_dS)NT��inplace)�superr�__init__r�conv1�NN�BatchNorm2d�bn1r�ReLU�relu�conv2�bn2�9downsampler��self�inplanes�planesrr'�	__class__s     �rrzBasicBlock.__init__s����
�j�$���(�(�*�*�*��X�v�v�6�6��10��>�&�)�)����G�D�)�)�)��	��V�V�,�,��11��>�&�)�)���$�������rc�H�|}|�|��}|�|��}|�|��}|�|��}|�|��}|j�|�|��}||z
}|�|��}|S�N)rr"r$r%r&r'�r)�x�residual�outs    r�forwardzBasicBlock.forward$s������j�j��m�m���h�h�s�m�m���i�i��n�n���j�j��o�o���h�h�s�m�m���?�&����q�)�)�H��x����i�i��n�n���12r�r	N��__name__�13__module__�__qualname__�	expansionrr3�
__classcell__�r,s@rrrsQ��������I�������������rrc�*��eZdZdZd�fd�	Zd�Z�xZS)�14Bottleneck�r	Nc���tt|�����tj||dd���|_t
j|��|_tj||d|dd���|_	t
j|��|_15tj|||jzdd���|_t
j||jz��|_
tjd���|_||_||_dS)Nr	F)r
rrrTr)rr=rrrrr r!r"r%r&r9�conv3�bn3r#r$r'rr(s     �rrzBottleneck.__init__:s����
�j�$���(�(�*�*�*��Y�x��Q�U�K�K�K��16��>�&�)�)����Y�v�v�1�V�'(�u�6�6�6��17��>�&�)�)����Y�v�v���'>�A�TY�Z�Z�Z��18��>�&�4�>�"9�:�:����G�D�)�)�)��	�$�������rc���|}|�|��}|�|��}|�|��}|�|��}|�|��}|�|��}|�|��}|�|��}|j�|�|��}||z
}|�|��}|Sr.)rr"r$r%r&r@rAr'r/s    rr3zBottleneck.forwardGs������j�j��m�m���h�h�s�m�m���i�i��n�n���j�j��o�o���h�h�s�m�m���i�i��n�n���j�j��o�o���h�h�s�m�m���?�&����q�)�)�H��x����i�i��n�n���19rr4r5r;s@rr=r=7sQ��������I�������������rr=c�.��eZdZd�fd�	Zdd�Zd�Z�xZS)r��c����d|_tt|�����t	jdddddd���|_tjd��|_	t	j20d���|_t	jddd	�21��|_
|�|d|d��|_|�|d|d	d�
��|_|�|d|dd�
��|_|�|d|dd�
��|_|���D]�}t)|tj��r(tj�|jdd����Dt)|tj��rJtj�|jd	��tj�|jd����dS)N�@r��FrTrr	)r
rrr�)r�i�fan_outr$)�mode�nonlinearity)r*rrrrrrr r!r"r#r$�	MaxPool2d�maxpool�_make_layer�layer1�layer2�layer3�layer4�modules�22isinstance�init�kaiming_normal_�weight�	constant_r)r)�block�layers�num_classes�mr,s     �rrzResNet.__init__`s������
�
�f�d���$�$�&�&�&��Y�q�"�!�A�q�$)�+�+�+��23��>�"�%�%����G�D�)�)�)��	��|��!�Q�G�G�G����&�&�u�b�&��)�<�<����&�&�u�c�6�!�9�Q�&�G�G����&�&�u�c�6�!�9�Q�&�G�G����&�&�u�c�6�!�9�Q�&�G�G��������	-�	-�A��!�R�Y�'�'�
-���'�'���y�v�'�V�V�V�V��A�r�~�.�.�
-���!�!�!�(�A�.�.�.���!�!�!�&�!�,�,�,��	-�	-rr	c	���d}|dks|j||jzkrStjtj|j||jzd|d���tj||jz����}g}|�||j|||����||jz|_td|��D]&}|�||j|�����'tj|�S)Nr	F)r
rr)	r*r9r�24Sequentialrr r!�append�range)r)r[r+�blocksrr'r\�is        rrPzResNet._make_layervs����25��Q�;�;�$�-�6�E�O�+C�C�C����	�$�-��%�/�)A�&'��U�D�D�D���v���7�8�8���J����
�
�e�e�D�M�6�6�:�F�F�G�G�G����0��
��q�&�!�!�	8�	8�A��M�M�%�%��
�v�6�6�7�7�7�7��}�f�%�%rc��g}|�|��}|�|��}|�|��}|�|��}|�|��}|�|��|�|��}|�|��|�|��}|�|��|�|��}|�|��|Sr.)	rr"r$rOrQrarRrSrT)r)r0�featuress   rr3zResNet.forward�s������J�J�q�M�M���H�H�Q�K�K���I�I�a�L�L���L�L��O�O���K�K��N�N����������K�K��N�N����������K�K��N�N����������K�K��N�N����������r)rD�r	)r6r7r8rrPr3r:r;s@rrr^s`�������-�-�-�-�-�-�,&�&�&�&�"������rrTc�0�ttgd�fi|��}|S)zxConstructs a ResNet-18 model.26    Args:27        pretrained (bool): If True, returns a model pre-trained on ImageNet28    )rHrHrHrH�rr��29pretrained�kwargs�models   rrr��$��30
�:�|�|�|�6�6�v�6�6�E��Lrc�0�ttgd�fi|��}|S)zxConstructs a ResNet-34 model.31    Args:32        pretrained (bool): If True, returns a model pre-trained on ImageNet33    �rr>�rrirjs   rrr�rnrc�0�ttgd�fi|��}|S)zxConstructs a ResNet-50 model.34    Args:35        pretrained (bool): If True, returns a model pre-trained on ImageNet36    rp�rr=rjs   rrr�s$��37
�:�|�|�|�6�6�v�6�6�E��Lrc�0�ttgd�fi|��}|S)zyConstructs a ResNet-101 model.38    Args:39        pretrained (bool): If True, returns a model pre-trained on ImageNet40    )rr>�rrsrjs   rrr�s$��41
�:�}�}�}�7�7��7�7�E��Lrc�0�ttgd�fi|��}|S)zyConstructs a ResNet-152 model.42    Args:43        pretrained (bool): If True, returns a model pre-trained on ImageNet44    )r��$rrsrjs   rrr�s$��45
�:�}�}�}�7�7��7�7�E��Lrrg)T)�torch.nnrr �__all__�46model_urlsr�Modulerr=rrrrrr�rr�<module>r~sa������������������47L�K�K�M�M���48�,�,�,�,�����������@$�$�$�$�$���$�$�$�N:�:�:�:�:�R�Y�:�:�:�z���������������������r