a 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*eee4efe5e5ee3d+d,d-Z6dS).N)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_v3csLeZdZeeddfdd ZeedddZd eeddd d Z Z S) QuantizableBasicConv2dNargskwargsreturncs tj|i|t|_dSN)super__init__nnZReLUreluselfrr __class__W/var/www/auris/lib/python3.9/site-packages/torchvision/models/quantization/inception.pyr!szQuantizableBasicConv2d.__init__xrcCs"||}||}||}|SrconvZbnr#)r%r+r(r(r)forwards   zQuantizableBasicConv2d.forwardis_qatrcCst|gd|dddS)Nr,T)Zinplace)r)r%r0r(r(r) fuse_model%sz!QuantizableBasicConv2d.fuse_model)N) __name__ __module__ __qualname__rr!rr.rboolr1 __classcell__r(r(r&r)rsrcs6eZdZeeddfdd ZeedddZZS)QuantizableInceptionANrcs&tj|dti|tj|_dSNZ conv_blockr r!rr" quantizedFloatFunctionalmyopr$r&r(r)r!+szQuantizableInceptionA.__init__r*cCs||}|j|dSNr_forwardr<catr%r+outputsr(r(r)r./s zQuantizableInceptionA.forwardr2r3r4rr!rr.r6r(r(r&r)r7)sr7cs6eZdZeeddfdd ZeedddZZS)QuantizableInceptionBNrcs&tj|dti|tj|_dSr8r9r$r&r(r)r!6szQuantizableInceptionB.__init__r*cCs||}|j|dSr=r>rAr(r(r)r.:s zQuantizableInceptionB.forwardrCr(r(r&r)rD4srDcs6eZdZeeddfdd ZeedddZZS)QuantizableInceptionCNrcs&tj|dti|tj|_dSr8r9r$r&r(r)r!AszQuantizableInceptionC.__init__r*cCs||}|j|dSr=r>rAr(r(r)r.Es zQuantizableInceptionC.forwardrCr(r(r&r)rE?srEcs6eZdZeeddfdd ZeedddZZS)QuantizableInceptionDNrcs&tj|dti|tj|_dSr8r9r$r&r(r)r!LszQuantizableInceptionD.__init__r*cCs||}|j|dSr=r>rAr(r(r)r.Ps zQuantizableInceptionD.forwardrCr(r(r&r)rFJsrFcsJeZdZeeddfdd ZeeedddZeeddd ZZ S) QuantizableInceptionENrcs>tj|dti|tj|_tj|_tj|_dSr8) 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 )Z kernel_sizeZstridepadding) branch1x1Z branch3x3_1Z branch3x3_2aZ branch3x3_2brHr@Zbranch3x3dbl_1Zbranch3x3dbl_2Zbranch3x3dbl_3aZbranch3x3dbl_3brIFZ avg_pool2d branch_pool)r%r+rLZ branch3x3Z branch3x3dblrNrBr(r(r)r?]s      zQuantizableInceptionE._forwardcCs||}|j|dSr=)r?rJr@rAr(r(r)r.rs zQuantizableInceptionE.forward) r2r3r4rr!rlistr?r.r6r(r(r&r)rGUsrGcs&eZdZeeddfdd ZZS)QuantizableInceptionAuxNrcstj|dti|dSr8)r r!rr$r&r(r)r!ysz QuantizableInceptionAux.__init__)r2r3r4rr!r6r(r(r&r)rPwsrPcsLeZdZeeddfdd ZeedddZd ee ddd d Z Z S) rNrc sDtj|dtttttttgi|t j j |_ t j j |_dS)NZinception_blocks)r r!rr7rDrErFrGrPtorchZaoZ quantizationZ QuantStubquantZ DeQuantStubdequantr$r&r(r)r!~s   zQuantizableInception3.__init__r*cCsj||}||}||\}}||}|jo6|j}tjrZ|sPt dt ||S| ||SdS)NzIScripted QuantizableInception3 always returns QuantizableInception3 Tuple) Z_transform_inputrRr?rSZtraining aux_logitsrQZjitZ 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)modulestyperr1)r%r0mr(r(r)r1s  z QuantizableInception3.fuse_model)N) r2r3r4rr!rr r.rr5r1r6r(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)KrZfbgemmzdhttps://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) r2r3r4rrr rr IMAGENET1K_V1IMAGENET1K_FBGEMM_V1DEFAULTr(r(r(r)rs( rZquantized_inception_v3)nameZ pretrainedcCs|ddrtjStjS)NquantizeF)getrrbrra)rr(r(r)s rg)weightsTF)rhprogressre)rhrirerrcKs|rtnt|}|dd}|durxd|vr:t|ddt|ddt|dt|jdd|jvrxt|d|jd|dd }tfi|}t ||rt |||dur|r|sd|_ d|_ | |j|dd |s|sd|_ d|_ |S) aInception v3 model architecture from `Rethinking the Inception Architecture for Computer Vision `__. .. 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: rTFNZtransform_inputTZ num_classesr\r]r[)riZ check_hash)rrverifyrfrlenr`poprrrrTZ AuxLogitsZload_state_dictZget_state_dict)rhrirerZoriginal_aux_logitsr]modelr(r(r)rs.4      r)7rU functoolsrtypingrrrrQZtorch.nnr"Ztorch.nn.functionalZ functionalrMrZtorchvision.modelsrZinception_moduleZtorchvision.models.inceptionrr Ztransforms._presetsr Z_apir rrZ_metar_utilsrrutilsrrr__all__Z BasicConv2drZ InceptionAr7Z InceptionBrDZ InceptionCrEZ InceptionDrFZ InceptionErGZ InceptionAuxrPZ Inception3rrr5rr(r(r(r)sN          ",