CoolFace
Apppublic

hololens/stable-diffusion-webui-depthmap-script

sourceHugging Faceupdated 2y agoView on Hugging Face
1likes
network_auxi.cpython-311.pyc68 linesDownload Raw Back to __pycache__
12��f�C��T�ddlZddlmZddlmcmZddlmZmZd�Zd�Z	Gd�dej3��ZGd�dej4��ZGd	�d5ej6��Z
Gd�dej7��ZGd
�dej8��ZGd�dej9��ZGd�dej10��ZGd�dej11��ZGd�dej12��ZedkrUedd���Zee��ejdddd��Zee��Zee�����dSdS)�N)�Resnet�
Resnext_torchc�*�tddgd����S)N�resnet�2��r	r	r	��backbone�depth�	upfactors��DepthNet���HF:\3dpicmaker\stable-diffusion-webui-depthmap-script\lib\network_auxi.py�resnet50_stride32rs���X�R�<�<�<�H�H�H�Hrc�*�tddgd����S)N�resnext101_32x8d�err13rrrr�resnext101_stride32x8drs���/�s�l�l�l�S�S�S�Src�*��eZdZ�fd�Zd�Zd�Z�xZS)�Decoderc���tt|�����gd�|_gd�|_gd�|_d|_t|jd|jd���|_tj14|jd|jddddd�	��|_tj|jdd15d���|_
t|jd|jd|jd|jd���|_t|jd|jd|jd|jd���|_t|jd
|jd
|jd
|jd
���|_t%|jd
|jd���|_|���dS)N���ii�rrrrr����16inchannels�midchannelsr	T��in_channels�out_channels�kernel_size�padding�stride�bias�bilinear��scale_factor�mode�
align_corners)r"r#�outchannels�upfactorr)r"r0r1)�superr�__init__r"r#r
r0�FTB�conv�nn�Conv2d�conv1�Upsample�upsample�FFM�ffm2�ffm1�ffm0�AO�outconv�_init_params)�self�	__class__s �rr3zDecoder.__init__s���
�g�t���%�%�'�'�'�1�1�1���/�/�/���"���������4�?�1�#5�4�CS�TU�CV�W�W�W��	��Y�4�+;�A�+>�T�M]�^_�M`�no�yz�DE�LP�Q�Q�Q��17������1B��cg�h�h�h��
��4�?�1�#5�4�CS�TU�CV�fj�fv�wx�fy�EI�ES�TU�EV�W�W�W��	��4�?�1�#5�4�CS�TU�CV�fj�fv�wx�fy�EI�ES�TU�EV�W�W�W��	��4�?�1�#5�4�CS�TU�CV�fj�fv�wx�fy�EI�ES�TU�EV�W�W�W��	��T�%5�a�%8�d�FV�ab�c�c�c����������rc���|���D�]W}t|tj��r=t	j|jd���|j�t	j|jd���Zt|tj	��r=t	j|jd���|j�t	j|jd����t|tj18��r6t	j|jd��t	j|jd����t|tj��r<t	j|jd���|j�t	j|jd����YdS�N�{�G�z�?��stdrr��modules�19isinstancer6r7�init�normal_�weightr*�	constant_�ConvTranspose2d�BatchNorm2d�Linear�rB�ms  rrAzDecoder._init_params"s;�������	.�	.�A��!�R�Y�'�'�
.���Q�X�4�0�0�0�0��6�%��N�1�6�1�-�-�-���A�r�1�2�2�20
.���Q�X�4�0�0�0�0��6�%��N�1�6�1�-�-�-���A�r�~�.�.�
.���q�x��+�+�+���q�v�q�)�)�)�)��A�r�y�)�)�
.���Q�X�4�0�0�0�0��6�%��N�1�6�1�-�-�-��	.�	.rc�b�|�|d��}|�|��}|�|��}|�|d|��}|�|d|��}|�|d|��}|�|��}|S)Nr r	rr)r5r8r:r<r=r>r@)	rB�features�x_32x�x_32�x_16�x_8�x_4�x_2�xs	         r�forwardzDecoder.forward4s����	�	�(�1�+�&�&���z�z�%� � ���}�}�T�"�"���i�i����T�*�*���i�i����S�)�)���i�i����S�)�)���L�L������r)�__name__�21__module__�__qualname__r3rAr^�
__classcell__�rCs@rrrsV������������$.�.�.�$22�23�24�25�26�27�28rrc�r��eZdZejejejejejd�Z	ddgd�f�fd�	Z29d�Z�xZS)r)��"rr�rrrc�30��tt|�����||_||_d|_gd�|_gd�|_||_d|_	|jdkrO|jtj31vrtd|j���tj32||j���|_dS|jdkr!tj|j���|_dStj|j���|_dS)	NFrrrrzUnsupported depth:)�33pretrainedr)r2rr3rrrir"r#r
r0�_DepthNet__factory�KeyError�encoderrr�34resnext101)rBrrr
rCs    �rr3zDepthNet.__init__Hs����	�h����&�&�(�(�(� ��
���35����0�0�0���/�/�/���"�������=�H�$�$��z��!3�3�3��3�T�Z�@�@�@�#�-�e�4���P�P�P�D�L�L�L�
�]�0�
0�
0�(�9�T�_�U�U�U�D�L�L�L�(�3�t��O�O�O�D�L�L�Lrc�0�|�|��}|S�N)rl�rBr]s  rr^zDepthNet.forward_s���L�L��O�O���r)
r_r`rar�resnet18�resnet34�resnet50�	resnet101�	resnet152rjr3r^rbrcs@rrr@s���������O��O��O�
�
�
�
���I�#��'�<�<�P�P�P�P�P�P�.������rrc�,��eZdZd�fd�	Zd�Zd�Z�xZS)r4rc�^��tt|�����||_||_tj|j|jdddd���|_tjtj	d���tj|j|jdddd���tj36|j���tj	d���tj|j|jdddd�����|_tj	d���|_|�
��dS)Nr rTr$��inplace��num_features)r2r4r3�in1�midr6r7r8�37Sequential�ReLUrQ�conv_branch�relu�init_params)rBr"r#rCs   �rr3zFTB.__init__es&���
�c�4���!�!�#�#�#��������Y�4�8�$�(�XY�cd�mn�$(�*�*�*��38��=����)>�)>�)>�)+��t�x�VZ�V^�lm�;<�Q�T�*S�*S�*S�)+��T�X�)N�)N�)N�)+���)>�)>�)>�)+��t�x�VZ�V^�lm�;<�Q�T�*S�*S�*S�T�T����G�D�)�)�)��	��������rc��|�|��}||�|��z}|�|��}|Sro)r8r�r�rps  rr^zFTB.forwardws=���J�J�q�M�M��
�� � ��#�#�#���I�I�a�L�L���rc���|���D�]W}t|tj��r=t	j|jd���|j�t	j|jd���Zt|tj	��r=t	j|jd���|j�t	j|jd����t|tj39��r6t	j|jd��t	j|jd����t|tj��r<t	j|jd���|j�t	j|jd����YdSrErIrSs  rr�zFTB.init_params~s;�������	.�	.�A��!�R�Y�'�'�
.���Q�X�4�0�0�0�0��6�%��N�1�6�1�-�-�-���A�r�1�2�2�
.���Q�X�4�0�0�0�0��6�%��N�1�6�1�-�-�-���A�r�~�.�.�
.���q�x��+�+�+���q�v�q�)�)�)�)��A�r�y�)�)�
.���Q�X�4�0�0�0�0��6�%��N�1�6�1�-�-�-��#	.�	.r)r�r_r`rar3r^r�rbrcs@rr4r4ds[�������������$���.�.�.�.�.�.�.rr4c�,��eZdZd�fd�	Zd�Zd�Z�xZS)�ATA�c����tt|�����||_t	jd��|_t	jt	j|jdz|j|z��t	j	d���t	j|j|z|j��t	j40����|_|���dS)Nrr	Trx)
r2r�r3r"r6�AdaptiveAvgPool2d�avg_poolr~rRr�Sigmoid�fcr�)rBr"�	reductionrCs   �rr3zATA.__init__�s����
�c�4���!�!�#�#�#�$����,�Q�/�/��
��-��	�$�/�A�*=�t��R[�?[� \� \� "��� 5� 5� 5� "�	�$�/�Y�*F��� X� X� "�41���.�.���	
�������rc�"�|���\}}}}tj||gd��}|�|��}|�|d��}|�|���||dd��}||z|z}|S)Nr�����)�size�torch�catr��viewr�)rB�low_x�high_x�n�c�_r]s       rr^zATA.forward�s����Z�Z�\�\�42��1�a���I�u�f�o�q�)�)���M�M�!����
�F�F�1�b�M�M���G�G�A�J�J�O�O�A�q�!�Q�'�'���A�I�����rc���|���D�]R}t|tj��r;t	j|j��|j�t	j|jd���Xt|tj	��r;t	j|j��|j�t	j|jd����t|tj43��r5t	j|jd��t	j|jd����t|tj��r<t	j|jd���|j�t	j|jd����TdS)NrrrFrG)
rJrKr6r7rL�xavier_normal_rNr*rOrPrQrRrMrSs  rr�zATA.init_params�s7�������	.�	.�A��!�R�Y�'�'�
.��#�A�H�-�-�-��6�%��N�1�6�1�-�-�-���A�r�1�2�2�
.��#�A�H�-�-�-��6�%��N�1�6�1�-�-�-���A�r�~�.�.�
.���q�x��+�+�+���q�v�q�)�)�)�)��A�r�y�)�)�
.���Q�X�4�0�0�0�0��6�%��N�1�6�1�-�-�-��'	.�	.r)r�r�rcs@rr�r��s[����������������.�.�.�.�.�.�.rr�c�,��eZdZd�fd�	Zd�Zd�Z�xZS)r;r	c�x��tt|�����||_||_||_||_t|j|j���|_t|j|j���|_	tj|jdd���|_|�
��dS)Nr!r+Tr,)r2r;r3r"r#r0r1r4�ftb1�ftb2r6r9r:r�)rBr"r#r0r1rCs     �rr3zFFM.__init__�s����
�c�4���!�!�#�#�#�$���&���&��� ��
��4�?��@P�Q�Q�Q��	��4�#3��AQ�R�R�R��	�����Z�_c�d�d�d��
��������rc��|�|��}||z}|�|��}|�|��}|Sro)r�r�r:)rBr�r�r]s    rr^zFFM.forward�sA���I�I�e����
��J���I�I�a�L�L���M�M�!�����rc���|���D�]W}t|tj��r=t	j|jd���|j�t	j|jd���Zt|tj	��r=t	j|jd���|j�t	j|jd����t|tj44��r6t	j|jd��t	j|jd����t|tj��r<t	j|jd���|j�t	j|jd����YdSrErIrSs  rr�zFFM.init_params��;�������	.�	.�A��!�R�Y�'�'�
.���Q�X�4�0�0�0�0��6�%��N�1�6�1�-�-�-���A�r�1�2�2�
.���Q�X�4�0�0�0�0��6�%��N�1�6�1�-�-�-���A�r�~�.�.�
.���q�x��+�+�+���q�v�q�)�)�)�)��A�r�y�)�)�
.���Q�X�4�0�0�0�0��6�%��N�1�6�1�-�-�-��'	.�	.r�r	r�rcs@rr;r;�s[�������
�
�
�
�
�
����.�.�.�.�.�.�.rr;c�,��eZdZd�fd�	Zd�Zd�Z�xZS)r?r	c
���tt|�����||_||_||_t
jt
j|j|jdzdddd���t
j	|jdz���t
j45d���t
j|jdz|jdddd���t
j|jdd�	����|_|�
��dS)46Nr	r rTr$rzrxr+r,)r2r?r3r"r0r1r6r~r7rQrr9�47adapt_convr�)rBr"r0r1rCs    �rr3zAO.__init__�s����
�b�$��� � �"�"�"�$���&��� ��
��-��I�$�/���ST�@T�bc�mn��T�
+�
+�
+��N���1�(<�=�=�=��G�D�!�!�!��I�$�/�Q�"6�T�EU�cd�no��T�
+�
+�
+��K�T�]��SW�X�X�X�Z�Z���	
�������rc�0�|�|��}|Sro)r�rps  rr^z48AO.forwards���O�O�A�����rc���|���D�]W}t|tj��r=t	j|jd���|j�t	j|jd���Zt|tj	��r=t	j|jd���|j�t	j|jd����t|tj49��r6t	j|jd��t	j|jd����t|tj��r<t	j|jd���|j�t	j|jd����YdSrErIrSs  rr�zAO.init_paramsr�rr�r�rcs@rr?r?�s[�������������"���.�.�.�.�.�.�.rr?c�*��eZdZ�fd�Zd�Zd�Z�xZS)�ResidualConvc���tt|�����tjtjd���tj||dzdddd���tj|dz���tjd���tj|dz|dddd�����|_|�	��dS)NFrxr	r rr$rz)50r2r�r3r6r~rr7rQr5r�)rBr"rCs  �rr3zResidualConv.__init__!s����
�l�D�!�!�*�*�,�,�,��M��G�E�"�"�"�
�I�*�:��>�WX�bc�lm� �
"�
"�
"��N�51�Q��7�7�7��G�E�"�"�"��I�*�q�.�z�WX�bc�lm� �
"�
"�
"�52�53��	�	
�������rc�6�|�|��|z}|Sro)r5rps  rr^zResidualConv.forward2s���I�I�a�L�L�1����rc���|���D�]W}t|tj��r=t	j|jd���|j�t	j|jd���Zt|tj	��r=t	j|jd���|j�t	j|jd����t|tj54��r6t	j|jd��t	j|jd����t|tj��r<t	j|jd���|j�t	j|jd����YdSrErIrSs  rr�zResidualConv.init_params6r�rr�rcs@rr�r� sV������������"���.�.�.�.�.�.�.rr�c�*��eZdZ�fd�Zd�Zd�Z�xZS)�
FeatureFusionc�P��tt|�����t|���|_tjt|���tj||dddd���tj|���tj	d�����|_55dS)	N)r"r r	r)r%r&r'r)r(�output_paddingrzTrx)r2r�r3r�r5r6r~rPrQr�up)rBr"r0rCs   �rr3zFeatureFusion.__init__Ns����
�m�T�"�"�+�+�-�-�-� �J�7�7�7��	��-��56� C� C� C� "� 2�z�Xc�qr�:;�Q�WX�!Z�!Z�!Z� "��K� H� H� H� "��� 5� 5� 5�	7�7����rc�X�|�||�|��z��Sro)r�r5)rB�lowfeat�highfeats   rr^zFeatureFusion.forwardXs%���w�w�x�$�)�)�G�"4�"4�4�5�5�5rc���|���D�]W}t|tj��r=t	j|jd���|j�t	j|jd���Zt|tj	��r=t	j|jd���|j�t	j|jd����t|tj57��r6t	j|jd��t	j|jd����t|tj��r<t	j|jd���|j�t	j|jd����YdSrErIrSs  rr�zFeatureFusion.init_params[r�rr�rcs@rr�r�MsV�������7�7�7�7�7�6�6�6�.�.�.�.�.�.�.rr�c�,��eZdZ�fd�Zd�Zdd�Z�xZS)�SenceUnderstandc�h��tt|�����||_t	jt	jdddd���t	jd�����|_t	j	d��|_58t	jt	jd|j��t	jd�����|_t	jt	j|j|jdd	���t	jd�����|_
|���dS)59Nrr r)r%r&r'r(Trxr�i�r)r2r�r3�channelsr6r~r7rr8r��poolrRr��conv2�initial_params)rBr�rCs  �rr3zSenceUnderstand.__init__ss����
�o�t�$�$�-�-�/�/�/� ��
��]�2�9��3�\]�gh�#i�#i�#i�#%�7�4�#8�#8�#8�:�:��60��(��+�+��	��-��	�+�t�}� E� E� "��� 5� 5� 5�7�7����]��I�$�-�d�m�YZ�de�f�f�f��G�D�!�!�!�#�#��61�	
�������rc�v�|���\}}}}|�|��}|�|��}|�|d��}|�|��}|�||jdd��}|�|��}|�dd||��}|S)Nr�r)r�r8r�r�r�r�r��repeat)rBr]r�r��h�ws      rr^zSenceUnderstand.forward�s����V�V�X�X�62��1�a���J�J�q�M�M���I�I�a�L�L��
�F�F�1�b�M�M���G�G�A�J�J��
�F�F�1�d�m�Q��*�*���J�J�q�M�M��
�H�H�Q��1�a� � ���rrFc�,�|���D]�}t|tj��rG|jj�d|��|j�|jj�d���ct|tj	��rG|jj�d|��|j�|jj�d����t|tj63��r |jj�d|����dS)Nr)rJrKr6r7rN�datarMr*�fill_rPrR)rB�devrTs   rr�zSenceUnderstand.initial_params�s��������	.�	.�A��!�R�Y�'�'�
.���
�%�%�a��-�-�-��6�%��F�K�%�%�a�(�(�(���A�r�1�2�2�
.���
�%�%�a��-�-�-��6�%��F�K�%�%�a�(�(�(���A�r�y�)�)�
.���
�%�%�a��-�-�-��	.�	.r)rF)r_r`rar3r^r�rbrcs@rr�r�rs[������������	�	�	�
.�
.�
.�
.�
.�
.�
.�
.rr��__main__rT)rri�r �)r��torch.nnr6�
torch.nn.initrL�librrrr�Modulerrr4r�r;r?r�r�r�r_�net�print�ones�inputs�outr�rrr�<module>r�s����������������������%�%�%�%�%�%�%�%�I�I�I�T�T�T�/�/�/�/�/�b�i�/�/�/�b!�!�!�!�!�r�y�!�!�!�H,.�,.�,.�,.�,.�"�)�,.�,.�,.�^).�).�).�).�).�"�)�).�).�).�X,.�,.�,.�,.�,.�"�)�,.�,.�,.�^+.�+.�+.�+.�+.���+.�+.�+.�d*.�*.�*.�*.�*.�2�9�*.�*.�*.�Z".�".�".�".�".�B�I�".�".�".�J&.�&.�&.�&.�&.�b�i�&.�&.�&.�R�z���64�(���65-�66-�67-�C�	�E�#�J�J�J�
�U�Z��!�C��
$�
$�F�68
�#�f�+�+�C�	�E�#�(�(�*�*�������r