hololens/stable-diffusion-webui-depthmap-script
1
1�
2 ��f( � �� � d dl mZ d dl mZ g d�Zdddddd�Zdd �Z G d3� d ej � Z G d � d
ej � Z G d� dej � Z dd�Z4dd�Z dd�Z dd�Z
dd�Zy)� 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 �8 � t j | |d|dd�� S )z 3x3 convolution with padding� � F�� kernel_size�stride�padding�bias)�nn�Conv2d)� in_planes�5out_planesr s �BF:\3dpicmaker\stable-diffusion-webui-depthmap-script\lib\Resnet.py�conv3x3r s � �
�9�9�Y�6��&� �U� ,� ,� c �* � � e Zd ZdZd� fd� Zd� Z� xZS )�7BasicBlockr c �* �� t t | � � t |||� | _ t j |� | _ t j d�� | _8 t ||� | _ t j |� | _ || _
|| _ y )NT��inplace)�superr �__init__r �conv1�NN� BatchNorm2d�bn1r �ReLU�relu�conv2�bn2�9downsampler ��self�inplanes�planesr r' � __class__s �r r zBasicBlock.__init__ sl �� �
�j�$�(�*� �X�v�v�6��10��>�>�&�)����G�G�D�)�� � �V�V�,��11��>�>�&�)���$��� �� r c � � |}| j |� }| j |� }| j |� }| j |� }| j |� }| j12 �| j |� }||z
}| j |� }|S �N)r r" r$ r% r& r' �r) �x�residual�outs r �forwardzBasicBlock.forward$ s{ � ����j�j��m���h�h�s�m���i�i��n���j�j��o���h�h�s�m�� �?�?� &����q�)�H� �x����i�i��n���13r �r N��__name__�14__module__� __qualname__� expansionr r3 �
__classcell__�r, s @r r r s �� ��I� �r r c �* � � e Zd ZdZd� fd� Zd� Z� xZS )�15Bottleneck� c � �� t t | � � t j ||dd�� | _ t
j |� | _ t j ||d|dd�� | _ t
j |� | _16 t j ||| j z dd�� | _ t
j || j z � | _
t j d�� | _ || _ || _ y )Nr F)r
r r17 r Tr )r r= r r r r r r! r" r% r& r9 �conv3�bn3r# r$ r' r r( s �r r zBottleneck.__init__: s� �� �
�j�$�(�*��Y�Y�x��Q�U�K��18��>�>�&�)����Y�Y�v�v�1�V�'(�u�6��19��>�>�&�)����Y�Y�v�v����'>�A�TY�Z��20��>�>�&�4�>�>�"9�:����G�G�D�)�� �$��� �� r c � � |}| j |� }| j |� }| j |� }| j |� }| j |� }| j |� }| j |� }| j
|� }| j �| j |� }||z
}| j |� }|S r. )r r" r$ r% r& r@ rA r' r/ s r r3 zBottleneck.forwardG s� � ����j�j��m���h�h�s�m���i�i��n���j�j��o���h�h�s�m���i�i��n���j�j��o���h�h�s�m�� �?�?� &����q�)�H� �x����i�i��n���21r r4 r5 r; s @r r= r= 7 s �� ��I� �r r= c �. � � e Zd Zd� fd� Zdd�Zd� Z� xZS )r c � �� d| _ t t | � � t j22 dddddd�� | _ t j d� | _ t j d�� | _ t j ddd �23� | _
| j |d|d � | _ | j |d |d d�
� | _ | j |d|d d�
� | _ | j |d|d d�
� | _ | j' � D ]� }t) |t j24 � r-t j* j- |j. dd�� �Jt) |t j � s�et j* j1 |j. d � t j* j1 |j2 d � �� y )N�@ r25 � � Fr 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�26isinstance�init�kaiming_normal_�weight� constant_r )r) �block�layers� num_classes�mr, s �r r zResNet.__init__` s_ �� ���
�
�f�d�$�&��Y�Y�q�"�!�A�q�$)�+��27��>�>�"�%����G�G�D�)�� ��|�|��!�Q�G�� ��&�&�u�b�&��)�<�� ��&�&�u�c�6�!�9�Q�&�G�� ��&�&�u�c�6�!�9�Q�&�G�� ��&�&�u�c�6�!�9�Q�&�G�� � � � � � -�A��!�R�Y�Y�'����'�'����y�v�'�V��A�r�~�~�.����!�!�!�(�(�A�.����!�!�!�&�&�!�,� -r c � � d }|dk7 s | j ||j z k7 ret j t j | j ||j z d|d�� t j ||j z � � }g }|j || j |||� � ||j z | _ t d|� D ]$ }|j || j |� � �& t j |� S )Nr F)r
r r ) r* r9 r �28Sequentialr r r! �append�range)r) rZ r+ �blocksr r' r[ �is r rO zResNet._make_layerv s� � ��29� �Q�;�$�-�-�6�E�O�O�+C�C�� � �� � �$�-�-��%�/�/�)A�&'��U�D����v���� 7�8��J� ���
�
�e�D�M�M�6�6�:�F�G� �� � �0��
��q�&�!� 8�A� �M�M�%��
�
�v�6� 7� 8� �}�}�f�%�%r c � � g }| j |� }| j |� }| j |� }| j |� }| j |� }|j |� | j
|� }|j |� | j |� }|j |� | j |� }|j |� |S r. ) r r" r$ rN rP r` rQ rR rS )r) r0 �featuress r r3 zResNet.forward� s� � ��� �J�J�q�M�� �H�H�Q�K�� �I�I�a�L�� �L�L��O�� �K�K��N������� �K�K��N������� �K�K��N������� �K�K��N��������r )i� �r )r6 r7 r8 r rO r3 r: r; s @r r r ^ s �� �-�,&�"r r c �, � t t g d�fi |��}|S )zxConstructs a ResNet-18 model.30 Args:31 pretrained (bool): If True, returns a model pre-trained on ImageNet32 )rG rG rG rG �r r ��33pretrained�kwargs�models r r r � � � �34
�:�|� 6�v� 6�E� �Lr c �, � t t g d�fi |��}|S )zxConstructs a ResNet-34 model.35 Args:36 pretrained (bool): If True, returns a model pre-trained on ImageNet37 �r38 r> � r39 rh ri s r r r � rm r c �, � t t g d�fi |��}|S )zxConstructs a ResNet-50 model.40 Args:41 pretrained (bool): If True, returns a model pre-trained on ImageNet42 ro �r r= ri s r r r � s � �43
�:�|� 6�v� 6�E� �Lr c �, � t t g d�fi |��}|S )zyConstructs a ResNet-101 model.44 Args:45 pretrained (bool): If True, returns a model pre-trained on ImageNet46 )r47 r> � r48 rr ri s r r r � s � �49
�:�}� 7�� 7�E� �Lr c �, � t t g d�fi |��}|S )zyConstructs a ResNet-152 model.50 Args:51 pretrained (bool): If True, returns a model pre-trained on ImageNet52 )r53 � �$ r54 rr ri s r r r � s � �55
�:�}� 7�� 7�E� �Lr rf )T)�torch.nnr r �__all__�56model_urlsr �Moduler r= r r r r r r � r r �<module>r} s| �� � � ��57 L�K�K�M�M� �58�,� ���� �@$���� $�N:�R�Y�Y� :�z����r