o hY* @sddlZddlmZddlmZmZmZddlZddlm Z ddl m m Z ddlm Z ddlmZddlmZmZddlmZd d lmZmZmZd d lmZd d lmZmZd dlm Z m!Z!m"Z"gdZ#Gdddej$Z%Gdddej&Z'Gdddej(Z)Gdddej*Z+Gdddej,Z-Gdddej.Z/Gdddej0Z1Gdddej2Z3Gd d!d!eZ4ed"d#ed$d%d&fd'dd(d)d*d+eee4efd,e5d-e5d.ed/e3f d0d1Z6dS)2N)partial)AnyOptionalUnion)Tensor) inception)Inception_V3_WeightsInceptionOutputs)ImageClassification)register_modelWeights WeightsEnum)_IMAGENET_CATEGORIES)_ovewrite_named_paramhandle_legacy_interface) _fuse_modules _replace_reluquantize_model)QuantizableInception3Inception_V3_QuantizedWeights inception_v3csTeZdZdededdffdd Zdedefdd Zd d eeddfd d Z Z S)QuantizableBasicConv2dargskwargsreturnNcs tj|i|t|_dSN)super__init__nnZReLUreluselfrr __class__o/home/www/facesmatcher.com/frenv_anti/lib/python3.10/site-packages/torchvision/models/quantization/inception.pyr szQuantizableBasicConv2d.__init__xcCs"||}||}||}|SrconvZbnr")r$r)r'r'r(forwards   zQuantizableBasicConv2d.forwardis_qatcCst|gd|dddS)Nr*T)Zinplace)r)r$r-r'r'r( fuse_model%sz!QuantizableBasicConv2d.fuse_modelr) __name__ __module__ __qualname__rr rr,rboolr. __classcell__r'r'r%r(rs rc<eZdZdededdffdd Zdedefdd ZZS) QuantizableInceptionArrrNc&tj|dti|tj|_dSNZ conv_blockrr rr! quantizedFloatFunctionalmyopr#r%r'r(r +zQuantizableInceptionA.__init__r)cC||}|j|dSNr_forwardr;catr$r)outputsr'r'r(r,/ zQuantizableInceptionA.forwardr/r0r1rr rr,r3r'r'r%r(r5)r5cr4) QuantizableInceptionBrrrNcr6r7r8r#r%r'r(r 6r<zQuantizableInceptionB.__init__r)cCr=r>r?rBr'r'r(r,:rDzQuantizableInceptionB.forwardrEr'r'r%r(rG4rFrGcr4) QuantizableInceptionCrrrNcr6r7r8r#r%r'r(r Ar<zQuantizableInceptionC.__init__r)cCr=r>r?rBr'r'r(r,ErDzQuantizableInceptionC.forwardrEr'r'r%r(rH?rFrHcr4) QuantizableInceptionDrrrNcr6r7r8r#r%r'r(r Lr<zQuantizableInceptionD.__init__r)cCr=r>r?rBr'r'r(r,PrDzQuantizableInceptionD.forwardrEr'r'r%r(rIJrFrIcsReZdZdededdffdd Zdedeefdd Zdedefd d ZZ S) QuantizableInceptionErrrNcs>tj|dti|tj|_tj|_tj|_dSr7) rr rr!r9r:myop1myop2myop3r#r%r'r(r Ws  zQuantizableInceptionE.__init__r)cCs||}||}||||g}|j|d}||}||}||| |g}|j |d}t j |dddd}| |}||||g}|S)Nrr )Z kernel_sizeZstridepadding) branch1x1Z branch3x3_1Z branch3x3_2aZ branch3x3_2brKrAZbranch3x3dbl_1Zbranch3x3dbl_2Zbranch3x3dbl_3aZbranch3x3dbl_3brLFZ avg_pool2d branch_pool)r$r)rOZ branch3x3Z branch3x3dblrQrCr'r'r(r@]s      zQuantizableInceptionE._forwardcCr=r>)r@rMrArBr'r'r(r,rrDzQuantizableInceptionE.forward) r/r0r1rr rlistr@r,r3r'r'r%r(rJUsrJcs*eZdZdededdffdd ZZS)QuantizableInceptionAuxrrrNcstj|dti|dSr7)rr rr#r%r'r(r ysz QuantizableInceptionAux.__init__)r/r0r1rr r3r'r'r%r(rSws"rScsTeZdZdededdffdd Zdedefdd Zd d ee ddfd d Z Z S)rrrrNc sDtj|dtttttttgi|t j j |_ t j j |_dS)NZinception_blocks)rr rr5rGrHrIrJrStorchZaoZ quantizationZ QuantStubquantZ DeQuantStubdequantr#r%r'r(r ~s   zQuantizableInception3.__init__r)cCsf||}||}||\}}||}|jo|j}tjr-|s(t dt ||S| ||S)NzIScripted QuantizableInception3 always returns QuantizableInception3 Tuple) Z_transform_inputrUr@rVZtraining aux_logitsrTZjitZ is_scriptingwarningswarnr Z eager_outputs)r$r)ZauxZ aux_definedr'r'r(r,s        zQuantizableInception3.forwardr-cCs(|D] }t|tur||qdS)aFuse conv/bn/relu modules in inception model Fuse conv+bn+relu/ conv+relu/conv+bn modules to prepare for quantization. Model is modified in place. Note that this operation does not change numerics and the model after modification is in floating point N)modulestyperr.)r$r-mr'r'r(r.s   z QuantizableInception3.fuse_modelr) r/r0r1rr rr r,rr2r.r3r'r'r%r(r}s rc@sJeZdZedeedddddeddejd d d d id ddd dZ e Z dS)rzUhttps://download.pytorch.org/models/quantized/inception_v3_google_fbgemm-a2837893.pthi+iV)Z crop_sizeZ resize_sizeir)Kr]fbgemmzdhttps://github.com/pytorch/vision/tree/main/references/classification#post-training-quantized-modelsz ImageNet-1Kg%CKS@g-VW@)zacc@1zacc@5g'1@gL7A`%7@z These weights were produced by doing Post Training Quantization (eager mode) on top of the unquantized weights listed below. ) Z num_paramsZmin_size categoriesbackendZrecipeZ unquantizedZ_metricsZ_ops _file_sizeZ_docs)urlZ transformsmetaN) r/r0r1rrr rr IMAGENET1K_V1IMAGENET1K_FBGEMM_V1DEFAULTr'r'r'r(rs* rZquantized_inception_v3)nameZ pretrainedcCs|ddr tjStjS)NquantizeF)getrrerrd)rr'r'r(s rj)weightsTF)rkprogressrhrkrlrhrrcKs|rtnt|}|dd}|dur`__. .. 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. .. note:: Note that ``quantize = True`` returns a quantized model with 8 bit weights. Quantized models only support inference and run on CPUs. GPU inference is not yet supported. Args: weights (:class:`~torchvision.models.quantization.Inception_V3_QuantizedWeights` or :class:`~torchvision.models.Inception_V3_Weights`, optional): The pretrained weights for the model. See :class:`~torchvision.models.quantization.Inception_V3_QuantizedWeights` 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. quantize (bool, optional): If True, return a quantized version of the model. Default is False. **kwargs: parameters passed to the ``torchvision.models.quantization.QuantizableInception3`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.quantization.Inception_V3_QuantizedWeights :members: .. autoclass:: torchvision.models.Inception_V3_Weights :members: :noindex: rWFNZtransform_inputTZ num_classesr_r`r^)rlZ check_hashr')rrverifyrirlenrcpoprrrrWZ AuxLogitsZload_state_dictZget_state_dict)rkrlrhrZoriginal_aux_logitsr`modelr'r'r(rs.4      r)7rX functoolsrtypingrrrrTZtorch.nnr!Ztorch.nn.functionalZ functionalrPrZtorchvision.modelsrZinception_moduleZtorchvision.models.inceptionrr Ztransforms._presetsr Z_apir rr_metarZ_utilsrrutilsrrr__all__Z BasicConv2drZ InceptionAr5Z InceptionBrGZ InceptionCrHZ InceptionDrIZ InceptionErJZ InceptionAuxrSZ Inception3rrr2rr'r'r'r(sX          ",