o V h?* @sddlZddlmZddlmZmZmZmZddlZddl m Z ddl m m Z ddlmZddlmZddlmZmZddlmZd d lmZmZmZd d lmZd d lmZmZd dl m!Z!m"Z"m#Z#gdZ$Gdddej%Z&Gdddej'Z(Gdddej)Z*Gdddej+Z,Gdddej-Z.Gdddej/Z0Gdddej1Z2Gdddej3Z4Gd d!d!eZ5ed"d#ed$d%d&fd'dd(d)d*d+eee5efd,e6d-e6d.ed/e4f d0d1Z7dS)2N)partial)AnyListOptionalUnion)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__nnReLUreluselfrr __class__]/var/www/vscode/kcb/lib/python3.10/site-packages/torchvision/models/quantization/inception.pyr!szQuantizableBasicConv2d.__init__xcCs"||}||}||}|Srconvbnr$)r&r+r)r)r*forwards   zQuantizableBasicConv2d.forwardis_qatcCst|gd|dddS)Nr,T)inplace)r)r&r0r)r)r* fuse_model%sz!QuantizableBasicConv2d.fuse_modelr) __name__ __module__ __qualname__rr!rr/rboolr2 __classcell__r)r)r'r*rs rc<eZdZdededdffdd Zdedefdd ZZS) QuantizableInceptionArrrNc&tj|dti|tj|_dSN conv_blockr r!rr" quantizedFloatFunctionalmyopr%r'r)r*r!+zQuantizableInceptionA.__init__r+cC||}|j|dSNr_forwardr@catr&r+outputsr)r)r*r// zQuantizableInceptionA.forwardr3r4r5rr!rr/r7r)r)r'r*r9)r9cr8) QuantizableInceptionBrrrNcr:r;r=r%r'r)r*r!6rAzQuantizableInceptionB.__init__r+cCrBrCrDrGr)r)r*r/:rIzQuantizableInceptionB.forwardrJr)r)r'r*rL4rKrLcr8) QuantizableInceptionCrrrNcr:r;r=r%r'r)r*r!ArAzQuantizableInceptionC.__init__r+cCrBrCrDrGr)r)r*r/ErIzQuantizableInceptionC.forwardrJr)r)r'r*rM?rKrMcr8) QuantizableInceptionDrrrNcr:r;r=r%r'r)r*r!LrAzQuantizableInceptionD.__init__r+cCrBrCrDrGr)r)r*r/PrIzQuantizableInceptionD.forwardrJr)r)r'r*rNJrKrNcsReZdZdededdffdd Zdedeefdd Zdedefd d ZZ S) QuantizableInceptionErrrNcs>tj|dti|tj|_tj|_tj|_dSr;) r r!rr"r>r?myop1myop2myop3r%r'r)r*r!Ws  zQuantizableInceptionE.__init__r+cCs||}||}||||g}|j|d}||}||}||| |g}|j |d}t j |dddd}| |}||||g}|S)Nrr ) kernel_sizestridepadding) branch1x1 branch3x3_1 branch3x3_2a branch3x3_2brPrFbranch3x3dbl_1branch3x3dbl_2branch3x3dbl_3abranch3x3dbl_3brQF avg_pool2d branch_pool)r&r+rV branch3x3 branch3x3dblr`rHr)r)r*rE]s      zQuantizableInceptionE._forwardcCrBrC)rErRrFrGr)r)r*r/rrIzQuantizableInceptionE.forward) r3r4r5rr!rrrEr/r7r)r)r'r*rOUsrOcs*eZdZdededdffdd ZZS)QuantizableInceptionAuxrrrNcstj|dti|dSr;)r r!rr%r'r)r*r!ysz QuantizableInceptionAux.__init__)r3r4r5rr!r7r)r)r'r*rcws"rccsTeZdZdededdffdd Zdedefdd Zd d ee ddfd d Z Z S)rrrrNc sDtj|dtttttttgi|t j j |_ t j j |_dS)Ninception_blocks)r r!rr9rLrMrNrOrctorchao quantization QuantStubquant DeQuantStubdequantr%r'r)r*r!~s   zQuantizableInception3.__init__r+cCsf||}||}||\}}||}|jo|j}tjr-|s(t dt ||S| ||S)NzIScripted QuantizableInception3 always returns QuantizableInception3 Tuple) _transform_inputrirErktraining aux_logitsrejit is_scriptingwarningswarnr eager_outputs)r&r+aux aux_definedr)r)r*r/s        zQuantizableInception3.forwardr0cCs(|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)modulestyperr2)r&r0mr)r)r*r2s   z QuantizableInception3.fuse_modelr) r3r4r5rr!rr r/rr6r2r7r)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) crop_size 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. ) num_paramsmin_size categoriesbackendrecipe unquantized_metrics_ops _file_size_docs)url transformsmetaN) r3r4r5rrr rr IMAGENET1K_V1IMAGENET1K_FBGEMM_V1DEFAULTr)r)r)r*rs* rquantized_inception_v3)name pretrainedcCs|ddr tjStjS)NquantizeF)getrrr r)rr)r)r*s r)weightsTF)rprogressrrrrrrcKs|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: rnFNtransform_inputT num_classesrrr|)r check_hashr))rr verifyrrlenrpoprrrrn AuxLogitsload_state_dictget_state_dict)rrrroriginal_aux_logitsrmodelr)r)r*rs.2      r)8rq functoolsrtypingrrrrretorch.nnr"torch.nn.functional functionalr^rtorchvision.modelsrinception_moduletorchvision.models.inceptionr r transforms._presetsr _apirrr_metar_utilsrrutilsrrr__all__ BasicConv2dr InceptionAr9 InceptionBrL InceptionCrM InceptionDrN InceptionErO InceptionAuxrc Inception3rrr6rr)r)r)r*sX          ",