a
    hI                     @   s  d dl Z d dlmZ d dlmZ d dlmZmZmZ d dl	Z	d dl
m  mZ d dl	mZmZ ddlmZ ddlmZ d	d
lmZmZmZ d	dlmZ d	dlmZmZ g dZedddgZeee de_eZG dd dej Z!G dd dej Z"G dd dej Z#G dd dej Z$G dd dej Z%G dd dej Z&G dd dej Z'G d d! d!ej Z(G d"d# d#eZ)e ed$e)j*fd%dd&d'ee) e+ee!d(d)d*Z,dS )+    N)
namedtuple)partial)AnyCallableOptional)nnTensor   )ImageClassification)_log_api_usage_once   )register_modelWeightsWeightsEnum)_IMAGENET_CATEGORIES)_ovewrite_named_paramhandle_legacy_interface)
Inception3InceptionOutputs_InceptionOutputsInception_V3_Weightsinception_v3r   logits
aux_logits)r   r   c                	       s   e Zd Zdeeeeeedej	f   ee e
dd fdd	Zeed
ddZeeeee f d
ddZejjeee edddZeed
ddZ  ZS )r     TFN      ?.)num_classesr   transform_inputinception_blocksinit_weightsdropoutreturnc                    s~  t    t|  |d u r,ttttttt	g}|d u rDt
dt d}t|dkrbtdt| |d }|d }|d }	|d }
|d	 }|d
 }|d }|| _|| _|ddddd| _|dddd| _|ddddd| _tjddd| _|dddd| _|dddd| _tjddd| _|ddd| _|ddd| _|ddd| _|	d| _|
ddd| _|
ddd| _|
ddd| _ |
ddd| _!d | _"|r|d|| _"|d| _#|d| _$|d| _%t&d| _'tj(|d| _)t*d|| _+|rz| , D ]}t-|tj.st-|tj*rHt/|dr(t0|j1nd}t2jj3j4|j5d |d!dd" n.t-|tj6rtj37|j5d tj37|j8d qd S )#NzThe default weight initialization of inception_v3 will be changed in future releases of torchvision. If you wish to keep the old behavior (which leads to long initialization times due to scipy/scipy#11299), please set init_weights=True.T   z2length of inception_blocks should be 7 instead of r   r   r	                   kernel_sizestrider)   @   r)   paddingP      )pool_features   i         )channels_7x7   i   i   r   r   )pstddevg?g        )meanZstdab)9super__init__r   BasicConv2d
InceptionA
InceptionB
InceptionC
InceptionD
InceptionEInceptionAuxwarningswarnFutureWarninglen
ValueErrorr   r   Conv2d_1a_3x3Conv2d_2a_3x3Conv2d_2b_3x3r   Z	MaxPool2dmaxpool1Conv2d_3b_1x1Conv2d_4a_3x3maxpool2Mixed_5bMixed_5cMixed_5dMixed_6aMixed_6bMixed_6cMixed_6dMixed_6e	AuxLogitsMixed_7aMixed_7bMixed_7cZAdaptiveAvgPool2davgpoolZDropoutr    Linearfcmodules
isinstanceConv2dhasattrfloatr9   torchinitZtrunc_normal_ZweightBatchNorm2dZ	constant_bias)selfr   r   r   r   r   r    
conv_blockZinception_aZinception_bZinception_cZinception_dZinception_eZinception_auxmr9   	__class__ J/var/www/auris/lib/python3.9/site-packages/torchvision/models/inception.pyr?      sj    	




zInception3.__init__xr!   c                 C   s|   | j rxt|d d df dd d }t|d d df dd d }t|d d df dd d	 }t|||fd}|S )
Nr   r   gZd;O?gQgy&1?gI+r	   g?gMbȿ)r   rg   Z	unsqueezecat)rk   rs   Zx_ch0Zx_ch1Zx_ch2rp   rp   rq   _transform_input_   s       zInception3._transform_inputc                 C   s  |  |}| |}| |}| |}| |}| |}| |}| |}| |}| 	|}| 
|}| |}| |}| |}| |}d }| jd ur| jr| |}| |}| |}| |}| |}| |}t|d}| |}||fS Nr   )rL   rM   rN   rO   rP   rQ   rR   rS   rT   rU   rV   rW   rX   rY   rZ   r[   trainingr\   r]   r^   r_   r    rg   flattenra   rk   rs   auxrp   rp   rq   _forwardg   s6    






















zInception3._forward)rs   rz   r!   c                 C   s   | j r| jrt||S |S d S )N)rw   r   r   ry   rp   rp   rq   eager_outputs   s    
zInception3.eager_outputsc                 C   sV   |  |}| |\}}| jo"| j}tj rF|s<td t	||S | 
||S d S )Nz3Scripted Inception3 always returns Inception3 Tuple)ru   r{   rw   r   rg   jitZis_scriptingrG   rH   r   r|   )rk   rs   rz   Zaux_definedrp   rp   rq   forward   s    



zInception3.forward)r   TFNNr   )__name__
__module____qualname__intboolr   listr   r   Modulerf   r?   r   ru   tupler{   rg   r}   Zunusedr   r|   r~   __classcell__rp   rp   rn   rq   r      s(         B6r   c                       s\   e Zd Zdeeeedejf  dd fddZe	e
e	 dddZe	e	dd	d
Z  ZS )rA   N.)in_channelsr1   rl   r!   c                    s   t    |d u rt}||ddd| _||ddd| _|ddddd| _||ddd| _|ddd	dd| _|ddd	dd| _|||dd| _	d S )
Nr,   r   r+   0   r%   r	   r-   `   r#   )
r>   r?   r@   	branch1x1branch5x5_1branch5x5_2branch3x3dbl_1branch3x3dbl_2branch3x3dbl_3branch_pool)rk   r   r1   rl   rn   rp   rq   r?      s    
zInceptionA.__init__rr   c                 C   sh   |  |}| |}| |}| |}| |}| |}tj|dddd}| |}||||g}|S Nr#   r   r)   r*   r.   )	r   r   r   r   r   r   F
avg_pool2dr   )rk   rs   r   Z	branch5x5branch3x3dblr   outputsrp   rp   rq   r{      s    






zInceptionA._forwardc                 C   s   |  |}t|dS rv   r{   rg   rt   rk   rs   r   rp   rp   rq   r~      s    
zInceptionA.forward)Nr   r   r   r   r   r   r   r   r?   r   r   r{   r~   r   rp   rp   rn   rq   rA      s    rA   c                       sZ   e Zd Zdeeedejf  dd fddZe	e
e	 dddZe	e	dd	d
Z  ZS )rB   N.r   rl   r!   c                    s`   t    |d u rt}||dddd| _||ddd| _|ddddd	| _|ddddd| _d S )
N  r#   r	   r(   r,   r   r+   r   r-   )r>   r?   r@   	branch3x3r   r   r   rk   r   rl   rn   rp   rq   r?      s    
zInceptionB.__init__rr   c                 C   sF   |  |}| |}| |}| |}tj|ddd}|||g}|S Nr#   r	   r(   )r   r   r   r   r   
max_pool2d)rk   rs   r   r   r   r   rp   rp   rq   r{      s    




zInceptionB._forwardc                 C   s   |  |}t|dS rv   r   r   rp   rp   rq   r~      s    
zInceptionB.forward)Nr   rp   rp   rn   rq   rB      s   &
rB   c                       s\   e Zd Zdeeeedejf  dd fddZe	e
e	 dddZe	e	dd	d
Z  ZS )rC   N.)r   r5   rl   r!   c                    s   t    |d u rt}||ddd| _|}|||dd| _|||ddd| _||dddd| _|||dd| _|||ddd| _|||ddd| _	|||ddd| _
||dddd| _||ddd| _d S )	Nr0   r   r+   r   r"   r   r#   r-   r"   r   r#   r   )r>   r?   r@   r   branch7x7_1branch7x7_2branch7x7_3branch7x7dbl_1branch7x7dbl_2branch7x7dbl_3branch7x7dbl_4branch7x7dbl_5r   )rk   r   r5   rl   Zc7rn   rp   rq   r?      s    
zInceptionC.__init__rr   c                 C   s   |  |}| |}| |}| |}| |}| |}| |}| |}| |}t	j
|dddd}| |}||||g}|S r   )r   r   r   r   r   r   r   r   r   r   r   r   )rk   rs   r   Z	branch7x7Zbranch7x7dblr   r   rp   rp   rq   r{   	  s    









zInceptionC._forwardc                 C   s   |  |}t|dS rv   r   r   rp   rp   rq   r~     s    
zInceptionC.forward)Nr   rp   rp   rn   rq   rC      s    rC   c                       sZ   e Zd Zdeeedejf  dd fddZe	e
e	 dddZe	e	dd	d
Z  ZS )rD   N.r   c                    s   t    |d u rt}||ddd| _|ddddd| _||ddd| _|dddd	d
| _|ddddd
| _|ddddd| _d S )Nr0   r   r+   @  r#   r	   r(   r   r   r-   r   r   )	r>   r?   r@   branch3x3_1branch3x3_2branch7x7x3_1branch7x7x3_2branch7x7x3_3branch7x7x3_4r   rn   rp   rq   r?   "  s    
zInceptionD.__init__rr   c                 C   sZ   |  |}| |}| |}| |}| |}| |}tj|ddd}|||g}|S r   )r   r   r   r   r   r   r   r   )rk   rs   r   Zbranch7x7x3r   r   rp   rp   rq   r{   .  s    






zInceptionD._forwardc                 C   s   |  |}t|dS rv   r   r   rp   rp   rq   r~   ;  s    
zInceptionD.forward)Nr   rp   rp   rn   rq   rD   !  s   &rD   c                       sZ   e Zd Zdeeedejf  dd fddZe	e
e	 dddZe	e	dd	d
Z  ZS )rE   N.r   c                    s   t    |d u rt}||ddd| _||ddd| _|ddddd| _|dddd	d| _||d
dd| _|d
dddd| _|ddddd| _	|dddd	d| _
||ddd| _d S )Nr   r   r+   r   )r   r#   )r   r   r-   )r#   r   )r   r   i  r#   r0   )r>   r?   r@   r   r   branch3x3_2abranch3x3_2br   r   branch3x3dbl_3abranch3x3dbl_3br   r   rn   rp   rq   r?   A  s    
zInceptionE.__init__rr   c                 C   s   |  |}| |}| || |g}t|d}| |}| |}| || 	|g}t|d}t
j|dddd}| |}||||g}|S )Nr   r#   r   )r   r   r   r   rg   rt   r   r   r   r   r   r   r   )rk   rs   r   r   r   r   r   rp   rp   rq   r{   R  s     




zInceptionE._forwardc                 C   s   |  |}t|dS rv   r   r   rp   rp   rq   r~   j  s    
zInceptionE.forward)Nr   rp   rp   rn   rq   rE   @  s   &rE   c                       sH   e Zd Zd	eeeedejf  dd fddZe	e	dddZ
  ZS )
rF   N.)r   r   rl   r!   c                    sX   t    |d u rt}||ddd| _|dddd| _d| j_td|| _d| j_d S )Nr4   r   r+   r3   r%   g{Gz?MbP?)	r>   r?   r@   conv0conv1r9   r   r`   ra   )rk   r   r   rl   rn   rp   rq   r?   p  s    
zInceptionAux.__init__rr   c                 C   sJ   t j|ddd}| |}| |}t |d}t|d}| |}|S )Nr%   r#   r(   r7   r   )r   r   r   r   Zadaptive_avg_pool2drg   rx   ra   rk   rs   rp   rp   rq   r~   |  s    


zInceptionAux.forward)N)r   r   r   r   r   r   r   r   r?   r   r~   r   rp   rp   rn   rq   rF   o  s    rF   c                       s8   e Zd Zeeedd fddZeedddZ  ZS )r@   N)r   out_channelskwargsr!   c                    s8   t    tj||fddi|| _tj|dd| _d S )Nrj   Fr   )Zeps)r>   r?   r   rd   convri   bn)rk   r   r   r   rn   rp   rq   r?     s    
zBasicConv2d.__init__rr   c                 C   s"   |  |}| |}tj|ddS )NT)Zinplace)r   r   r   Zrelur   rp   rp   rq   r~     s    

zBasicConv2d.forward)	r   r   r   r   r   r?   r   r~   r   rp   rp   rn   rq   r@     s   r@   c                   @   sD   e Zd Zedeedddddeddd	d
didddddZeZdS )r   zDhttps://download.pytorch.org/models/inception_v3_google-0cc3c7bd.pthi+  iV  )Z	crop_sizeZresize_sizeir)K   r   zRhttps://github.com/pytorch/vision/tree/main/references/classification#inception-v3zImageNet-1KgA`RS@g\W@)zacc@1zacc@5g'1@goY@z1These weights are ported from the original paper.)Z
num_paramsZmin_size
categoriesZrecipeZ_metricsZ_ops
_file_sizeZ_docs)urlZ
transformsmetaN)	r   r   r   r   r   r
   r   IMAGENET1K_V1DEFAULTrp   rp   rp   rq   r     s$   r   Z
pretrained)weightsT)r   progress)r   r   r   r!   c                 K   s   t | } |dd}| dur`d|vr2t|dd t|dd t|dd t|dt| jd  tf i |}| dur|| j|dd	 |sd|_	d|_
|S )
ak  
    Inception v3 model architecture from
    `Rethinking the Inception Architecture for Computer Vision <http://arxiv.org/abs/1512.00567>`_.

    .. note::
        **Important**: In contrast to the other models the inception_v3 expects tensors with a size of
        N x 3 x 299 x 299, so ensure your images are sized accordingly.

    Args:
        weights (:class:`~torchvision.models.Inception_V3_Weights`, optional): The
            pretrained weights for the model. See
            :class:`~torchvision.models.Inception_V3_Weights` below for
            more details, and possible values. By default, no pre-trained
            weights are used.
        progress (bool, optional): If True, displays a progress bar of the
            download to stderr. Default is True.
        **kwargs: parameters passed to the ``torchvision.models.Inception3``
            base class. Please refer to the `source code
            <https://github.com/pytorch/vision/blob/main/torchvision/models/inception.py>`_
            for more details about this class.

    .. autoclass:: torchvision.models.Inception_V3_Weights
        :members:
    r   TNr   r   Fr   r   )r   Z
check_hash)r   verifygetr   rJ   r   r   Zload_state_dictZget_state_dictr   r[   )r   r   r   Zoriginal_aux_logitsmodelrp   rp   rq   r     s    
r   )-rG   collectionsr   	functoolsr   typingr   r   r   rg   Ztorch.nn.functionalr   Z
functionalr   r   Ztransforms._presetsr
   utilsr   Z_apir   r   r   Z_metar   _utilsr   r   __all__r   __annotations__r   r   r   rA   rB   rC   rD   rE   rF   r@   r   r   r   r   rp   rp   rp   rq   <module>   s8    './