a dI@sddlZddlmZddlmZddlmZmZmZm Z m Z ddl Z ddl m mZddl m Z mZddlmZddlmZd d lmZmZmZd d lmZd d lmZmZgd ZedddgZee ede_ eZ!Gddde j"Z#Gddde j"Z$Gddde j"Z%Gddde j"Z&Gddde j"Z'Gddde j"Z(Gddde j"Z)Gd d!d!e j"Z*Gd"d#d#eZ+eed$e+j,fd%dd&d'e e+e-ee#d(d)d*Z.dS)+N) namedtuple)partial)AnyCallableListOptionalTuple)nnTensor)ImageClassification)_log_api_usage_once)register_modelWeights WeightsEnum)_IMAGENET_CATEGORIES)_ovewrite_named_paramhandle_legacy_interface) Inception3InceptionOutputs_InceptionOutputsInception_V3_Weights inception_v3rlogits aux_logits)rrc seZdZdeeeeeedej feee ddfdd Z e e d d d Z e ee ee fd d dZejje ee edddZe ed ddZZS)rTFN?.) num_classesrtransform_inputinception_blocks init_weightsdropoutreturncs~tt||dur,ttttttt g}|durDt dt d}t |dkrbtdt ||d}|d}|d} |d} |d } |d } |d } ||_||_|dd ddd |_|d d dd|_|d dddd|_tjddd |_|dddd|_|dddd|_tjddd |_|dd d|_|ddd|_|ddd|_| d|_| ddd|_| ddd|_| ddd|_ | ddd|_!d|_"|r| d||_"| d|_#| d|_$| d|_%t&d|_'tj(|d|_)t*d||_+|rz|,D]}t-|tj.st-|tj*rHt/|dr(t0|j1nd}t2jj3j4|j5d |d!dd"n.t-|tj6rtj37|j5dtj37|j8dqdS)#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.Tz2length of inception_blocks should be 7 instead of rrr   kernel_sizestrider+@r+paddingP) pool_featuresi ) channels_7x7iirr)pstddevg?g)meanZstdab)9super__init__r BasicConv2d InceptionA InceptionB InceptionC InceptionD InceptionE InceptionAuxwarningswarn FutureWarninglen ValueErrorrr Conv2d_1a_3x3 Conv2d_2a_3x3 Conv2d_2b_3x3r Z MaxPool2dmaxpool1 Conv2d_3b_1x1 Conv2d_4a_3x3maxpool2Mixed_5bMixed_5cMixed_5dMixed_6aMixed_6bMixed_6cMixed_6dMixed_6e AuxLogitsMixed_7aMixed_7bMixed_7cZAdaptiveAvgPool2davgpoolZDropoutr"Linearfcmodules isinstanceConv2dhasattrfloatr;torchinitZ trunc_normal_Zweight BatchNorm2dZ constant_bias)selfrrrr r!r" conv_blockZ inception_aZ inception_bZ inception_cZ inception_dZ inception_eZ inception_auxmr; __class__e/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torchvision/models/inception.pyrAsj         zInception3.__init__xr#cCs||jrxt|dddfddd}t|dddfddd}t|dddfddd }t|||fd}|S) NrrgZd;O?gQgy&1?gI +r g?gMbȿ)rriZ unsqueezecat)rmruZx_ch0Zx_ch1Zx_ch2rrrrrs_transform_input_s    zInception3._transform_inputcCs||}||}||}||}||}||}||}||}||}| |}| |}| |}| |}| |}||}d}|jdur|jr||}||}||}||}||}||}t|d}||}||fSNr)rNrOrPrQrRrSrTrUrVrWrXrYrZr[r\r]trainingr^r_r`rar"riflattenrcrmruauxrrrrrs_forwardgs6                        zInception3._forward)rur|r#cCs|jr|jrt||S|SdS)N)ryrrr{rrrrrs eager_outputss  zInception3.eager_outputscCsV||}||\}}|jo"|j}tjrF|s`_. .. 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 `_ for more details about this class. .. autoclass:: torchvision.models.Inception_V3_Weights :members: rTNrr!Frr)r) rverifygetrrLrrZload_state_dictZget_state_dictrr])rrrZoriginal_aux_logitsmodelrrrrrsrs     r)/rI collectionsr functoolsrtypingrrrrrriZtorch.nn.functionalr Z functionalrr Ztransforms._presetsr utilsr Z_apirrrZ_metar_utilsrr__all__r__annotations__rrrrCrDrErFrGrHrBrrrrrrrrrrrss8     './