U h?* @sddlZddlmZddlmZmZmZmZddlZddl m Z ddl m m Z ddlmZddlmZddlmZmZddlmZd d lmZmZmZd d lmZd d lmZmZd dl m!Z!m"Z"m#Z#dddgZ$Gdddej%Z&Gdddej'Z(Gdddej)Z*Gdddej+Z,Gdddej-Z.Gdddej/Z0Gdddej1Z2Gd ddej3Z4Gd!ddeZ5ed"d#ed$d%d&fd'dd(d)d*eee5efe6e6ee4d+d,dZ7dS)-N)partial)AnyListOptionalUnion)Tensor) inception)Inception_V3_WeightsInceptionOutputs)ImageClassification)register_modelWeights WeightsEnum)_IMAGENET_CATEGORIES)_ovewrite_named_paramhandle_legacy_interface) _fuse_modules _replace_reluquantize_modelQuantizableInception3Inception_V3_QuantizedWeights inception_v3csLeZdZeeddfdd ZeedddZd eeddd d Z Z S) QuantizableBasicConv2dNargskwargsreturncstj||t|_dSN)super__init__nnZReLUreluselfrr __class__[/var/www/html/venv/lib/python3.8/site-packages/torchvision/models/quantization/inception.pyr"szQuantizableBasicConv2d.__init__xrcCs"||}||}||}|Sr )convbnr$)r&r,r)r)r*forwards   zQuantizableBasicConv2d.forwardis_qatrcCst|dddg|dddS)Nr-r.r$T)Zinplace)r)r&r1r)r)r* fuse_model%sz!QuantizableBasicConv2d.fuse_model)N) __name__ __module__ __qualname__rr"rr/rboolr2 __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.forwardr3r4r5rr"rr/r7r)r)r'r*r8)sr8cs6eZdZeeddfdd ZeedddZZS)QuantizableInceptionBNrcs&tj|dti|tj|_dSr9r:r%r'r)r*r"6szQuantizableInceptionB.__init__r+cCs||}|j|dSr>r?rBr)r)r*r/:s zQuantizableInceptionB.forwardrDr)r)r'r*rE4srEcs6eZdZeeddfdd ZeedddZZS)QuantizableInceptionCNrcs&tj|dti|tj|_dSr9r:r%r'r)r*r"AszQuantizableInceptionC.__init__r+cCs||}|j|dSr>r?rBr)r)r*r/Es zQuantizableInceptionC.forwardrDr)r)r'r*rF?srFcs6eZdZeeddfdd ZeedddZZS)QuantizableInceptionDNrcs&tj|dti|tj|_dSr9r:r%r'r)r*r"LszQuantizableInceptionD.__init__r+cCs||}|j|dSr>r?rBr)r)r*r/Ps zQuantizableInceptionD.forwardrDr)r)r'r*rGJsrGcsJeZdZeeddfdd ZeeedddZeeddd ZZ S) QuantizableInceptionENrcs>tj|dti|tj|_tj|_tj|_dSr9) 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_2brIrAZbranch3x3dbl_1Zbranch3x3dbl_2Zbranch3x3dbl_3aZbranch3x3dbl_3brJFZ avg_pool2d branch_pool)r&r,rMZ branch3x3Z branch3x3dblrOrCr)r)r*r@]s      zQuantizableInceptionE._forwardcCs||}|j|dSr>)r@rKrArBr)r)r*r/rs zQuantizableInceptionE.forward) r3r4r5rr"rrr@r/r7r)r)r'r*rHUsrHcs&eZdZeeddfdd ZZS)QuantizableInceptionAuxNrcstj|dti|dSr9)r!r"rr%r'r)r*r"ysz QuantizableInceptionAux.__init__)r3r4r5rr"r7r)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"rr8rErFrGrHrPtorchZaoZ 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.forwardr0cCs(|D]}t|tkr||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&r1mr)r)r*r2s  z QuantizableInception3.fuse_model)N) r3r4r5rr"rr r/rr6r2r7r)r)r'r*r}s c@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) r3r4r5rrr rr IMAGENET1K_V1IMAGENET1K_FBGEMM_V1DEFAULTr)r)r)r*rs( Zquantized_inception_v3)nameZ pretrainedcCs|ddrtjStjS)NquantizeF)getrrbr ra)rr)r)r*s rg)weightsTF)rhprogressre)rhrirerrcKs|rtnt|}|dd}|dk rxd|kr:t|ddt|ddt|dt|jdd|jkrxt|d|jd|dd }tf|}t ||rt |||dk r|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)rr verifyrfrlenr`poprrrrTZ AuxLogitsZload_state_dictZget_state_dict)rhrirerZoriginal_aux_logitsr]modelr)r)r*rs.2       )8rU functoolsrtypingrrrrrQZtorch.nnr#Ztorch.nn.functionalZ functionalrNrZtorchvision.modelsrZinception_moduleZtorchvision.models.inceptionr r Ztransforms._presetsr Z_apirrr_metarZ_utilsrrutilsrrr__all__Z BasicConv2drZ InceptionAr8Z InceptionBrEZ InceptionCrFZ InceptionDrGZ InceptionErHZ InceptionAuxrPZ Inception3rrr6rr)r)r)r*sT          ",