CoolFace
Apppublic

hololens/stable-diffusion-webui-depthmap-script

sourceHugging Faceupdated 2y agoView on Hugging Face
1likes
Resnext_torch.cpython-311.pyc70 linesDownload Raw Back to __pycache__
12��f#���ddlmZ	ddlmZn#e$r	ddlmZYnwxYwdgZddd�Zdd�Zdd	�Z	Gd3�dej4��ZGd�d
ej5��ZGd�dej6��Z
dd�ZedkrdddlZed�����Zejd�����Zee��Zeee����dSdS)�N)�urlretrieve�resnext101_32x8dz@https://download.pytorch.org/models/resnext50_32x4d-7cdf4587.pthzAhttps://download.pytorch.org/models/resnext101_32x8d-8ba56ff5.pth)�resnext50_32x4dr�c7�:�tj||d|||d|���S)z3x3 convolution with padding�F)�kernel_size�stride�padding�groups�bias�dilation��nn�Conv2d)�	in_planes�8out_planesr9rrs     �IF:\3dpicmaker\stable-diffusion-webui-depthmap-script\lib\Resnext_torch.py�conv3x3rs0��
�9�Y�10��&�%�f�5�8�U�U�U�U�c�4�tj||d|d���S)z1x1 convolutionrF)r	r11r
r)rrr12s   r�conv1x1rs��
�9�Y�13��&�u�U�U�U�Urc�.��eZdZdZ		d�fd�	Zd�Z�xZS)�14BasicBlockrN�@c	���tt|�����|�tj}|dks|dkrtd���|dkrt
d���t|||��|_||��|_	tj15d���|_t||��|_||��|_
||_||_dS)Nrrz3BasicBlock only supports groups=1 and base_width=64z(Dilation > 1 not supported in BasicBlockT��inplace)�superr�__init__r�BatchNorm2d�16ValueError�NotImplementedErrorr�conv1�bn1�ReLU�relu�conv2�bn2�17downsampler18)19�self�inplanes�planesr20r*r�21base_widthr�22norm_layer�	__class__s23         �rr zBasicBlock.__init__!s����
�j�$���(�(�*�*�*�����J��Q�;�;�*��*�*��R�S�S�S��a�<�<�%�&P�Q�Q�Q��X�v�v�6�6��24��:�f�%�%����G�D�)�)�)��	��V�V�,�,��25��:�f�%�%���$�������rc�H�|}|�|��}|�|��}|�|��}|�|��}|�|��}|j�|�|��}||z
}|�|��}|S�N)r$r%r'r(r)r*�r+�x�identity�outs    r�forwardzBasicBlock.forward3s������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���26r�rNrrrN��__name__�27__module__�__qualname__�	expansionr r7�
__classcell__�r0s@rrrsW��������I�KL�7;�������$������rrc�.��eZdZdZ		d�fd�	Zd�Z�xZS)�28Bottleneck�rNrc	����tt|�����|�tj}t||dzz��|z}	t
||	��|_||	��|_t|	|	|||��|_29||	��|_t
|	||jz��|_
|||jz��|_tjd���|_||_||_dS)NgP@Tr)rrAr rr!�intrr$r%rr(r)r=�conv3�bn3r&r'r*r30)r+r,r-r31r*rr.rr/�widthr0s          �rr zBottleneck.__init__Os����
�j�$���(�(�*�*�*�����J��F�j�3�.�/�0�0�6�9���X�u�-�-��32��:�e�$�$����U�E�6�6�8�D�D��33��:�e�$�$����U�F�T�^�$;�<�<��34��:�f�t�~�5�6�6����G�D�)�)�)��	�$�������rc���|}|�|��}|�|��}|�|��}|�|��}|�|��}|�|��}|�|��}|�|��}|j�|�|��}||z
}|�|��}|Sr2)r$r%r'r(r)rErFr*r3s    rr7zBottleneck.forward`s������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���35rr8r9r?s@rrArAFsY��������I�KL�7;�������"������rrAc�:��eZdZ			d36�fd�	Zdd�Zd�Zd	�Z�xZS)�ResNet��FrrNc	���tt|�����|�tj}||_d|_d|_|�gd�}t|��dkr"td�37|�����||_||_tj
d|jdddd�	��|_||j��|_tjd38���|_tjddd���|_|�|d|d
��|_|�|d|dd|d
���|_|�|d|dd|d���|_|�|d|dd|d���|_|���D]�}	t5|	tj
��r(tj�|	jdd����Dt5|	tjtjf��rJtj�|	jd��tj�|	j d
����|r�|���D]�}	t5|	tB��r+tj�|	j"jd
���Bt5|	tF��r*tj�|	j$jd
����dSdS)Nrr)FFFrzHreplace_stride_with_dilation should be None or a 3-element tuple, got {}��F)r	r39rr
Tr)r	r40rr�)r41�dilate�i�fan_outr')�mode�nonlinearity)%rrJr rr!�_norm_layerr,r�lenr"�formatrr.rr$r%r&r'�	MaxPool2d�maxpool�_make_layer�layer1�layer2�layer3�layer4�modules�42isinstance�init�kaiming_normal_�weight�	GroupNorm�	constant_r
rArFrr))r+�block�layers�num_classes�zero_init_residualr�width_per_group�replace_stride_with_dilationr/�mr0s          �rr zResNet.__init__ys����	�f�d���$�$�&�&�&�����J�%�����
���
�'�/�,A�+@�+@�(��+�,�,��1�1��<�<B�F�C_�<`�<`�b�b�
b����)����Y�q�$�-�Q�q�RS�$)�+�+�+��43��:�d�m�,�,����G�D�)�)�)��	��|��!�Q�G�G�G����&�&�u�b�&��)�<�<����&�&�u�c�6�!�9�Q�.J�1�.M�'�O�O����&�&�u�c�6�!�9�Q�.J�1�.M�'�O�O����&�&�u�c�6�!�9�Q�.J�1�.M�'�O�O���44�����	-�	-�A��!�R�Y�'�'�
-���'�'���y�v�'�V�V�V�V��A�����=�>�>�
-���!�!�!�(�A�.�.�.���!�!�!�&�!�,�,�,��45�	7��\�\�^�^�
7�
7���a��,�,�7��G�%�%�a�e�l�A�6�6�6�6���:�.�.�7��G�%�%�a�e�l�A�6�6�6��	7�	7�
7�
7rc�<�|j}d}|j}|r|xj|zc_d}|dks|j||jzkrBt	jt
|j||jz|��|||jz����}g}	|	�||j||||j|j	||����||jz|_td|��D]:}46|	�||j||j|j	|j|������;t	j|	�S)Nr)rr.rr/)rUrr,r=r�47Sequentialr�appendrr.�range)r+rfr-�blocksr48rPr/r*�previous_dilationrg�_s           rrZzResNet._make_layer�sO���%�49��50� �M���	��M�M�V�#�M�M��F��Q�;�;�$�-�6�E�O�+C�C�C�����
�v���'?��H�H��51�6�E�O�3�4�4���J�52���
�
�e�e�D�M�6�6�:�t�{� �O�->�53�L�L�	M�	M�	M����0��
��q�&�!�!�	8�	8�A��M�M�%�%��
�v�d�k�+/�?�T�]�+5�7�7�7�
8�
8�
8�
8��}�f�%�%rc��g}|�|��}|�|��}|�|��}|�|��}|�|��}|�|��|�|��}|�|��|�|��}|�|��|�|��}|�|��|Sr2)	r$r%r'rYr[ror\r]r^)r+r4�featuress   r�
_forward_implzResNet._forward_impl�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����������rc�,�|�|��Sr2)rv)r+r4s  rr7zResNet.forward�s���!�!�!�$�$�$r)rKFrrNN)rF)r:r;r<r rZrvr7r>r?s@rrJrJwsy�������KP�LP� �17�17�17�17�17�17�f&�&�&�&�0���4%�%�%�%�%�%�%rrJTc�D�d|d<d|d<ttgd�fi|��}|S)zyConstructs a ResNet-152 model.54    Args:55        pretrained (bool): If True, returns a model pre-trained on ImageNet56    � r�rj)rrB�r)rJrA)�57pretrained�kwargs�models   rrr�s9��58�F�8�� !�F����:�}�}�}�7�7��7�7�E��Lr�__main__)rNrrQrQ)rrr)r)T)�torch.nnr�urllibr�ImportError�urllib.request�__all__�59model_urlsrr�ModulerrArJrr:�torch�cudar~�rand�rgbr6�printrV�rr�<module>r�s���������+�"�"�"�"�"�"�"���+�+�+�*�*�*�*�*�*�*�*�+�����60��Z�[���61�U�U�U�U�V�V�V�V�62%�%�%�%�%���%�%�%�P.�.�.�.�.���.�.�.�bh%�h%�h%�h%�h%�R�Y�h%�h%�h%�X	�	�	�	��z����L�L�L���T�"�"�'�'�)�)�E�63�%�*�%�64&�65&�66+�67+�68-�69-�C�70�%��*�*�C�	�E�#�#�c�(�(�O�O�O�O�O�
�s���