U (d+@sddlZddlmZddlmZmZmZmZddlZddl m Z ddl m m Z ddlmZddlmZddlmZmZddlmZd d lmZmZd d lmZd d lmZmZd dlm Z m!Z!m"Z"dddgZ#Gdddej$Z%Gdddej&Z'Gdddej(Z)Gdddej*Z+Gdddej,Z-Gdddej.Z/Gdddej0Z1Gd ddej2Z3Gd!ddeZ4ed"d#d$fd%dd&d'd(eee4efe5e5ee3d)d*dZ6d d+lm7Z7d d,lm8Z8e7d-e4j9j:iZ;dS).N)partial)AnyListOptionalUnion)Tensor) inception)InceptionOutputsInception_V3_Weights)ImageClassification) WeightsEnumWeights)_IMAGENET_CATEGORIES)handle_legacy_interface_ovewrite_named_param) _fuse_modules _replace_reluquantize_modelQuantizableInception3Inception_V3_QuantizedWeights inception_v3csLeZdZeeddfdd ZeedddZd eeddd d Z Z S) QuantizableBasicConv2dNargskwargsreturncstj||t|_dSN)super__init__nnZReLUreluselfrr __class__M/tmp/pip-unpacked-wheel-vx7f76es/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%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!rrr?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)rOwsrOcsPeZdZdeeeddfdd Zeedd d Zde edd d d Z Z S)rTFN) num_classes aux_logitstransform_inputrc sBtj|||tttttttgdt j j |_ t j j |_dS)N)rQrRrSZinception_blocks)r r!rr7rDrErFrGrOtorchZaoZ quantizationZ QuantStubquantZ DeQuantStubdequant)r%rQrRrSr&r(r)r!~szQuantizableInception3.__init__r*cCsj||}||}||\}}||}|jo6|j}tjrZ|sPt dt ||S| ||SdS)NzIScripted QuantizableInception3 always returns QuantizableInception3 Tuple) Z_transform_inputrUr?rVZtrainingrRrTZjitZ is_scriptingwarningswarnr Z eager_outputs)r%r+ZauxZ aux_definedr(r(r)r.s       zQuantizableInception3.forwardr/cCs(|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)modulestyperr1)r%r0mr(r(r)r1s  z QuantizableInception3.fuse_model)rPTF)N) r2r3r4intr5r!rr r.rr1r6r(r(r&r)r}s c @sFeZdZedeedddddeddejd d d d id ddZ e Z dS)rzUhttps://download.pytorch.org/models/quantized/inception_v3_google_fbgemm-71447a44.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@5z 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_docs)urlZ transformsmetaN) r2r3r4rrr rr IMAGENET1K_V1IMAGENET1K_FBGEMM_V1DEFAULTr(r(r(r)rs$ Z pretrainedcCs|ddrtjStjS)NquantizeF)getrrdr rc)rr(r(r)s rh)weightsTF)riprogressrf)rirjrfrrcKs|rtnt|}|dd}|dk rxd|kr:t|ddt|ddt|dt|jdd|jkrxt|d|jd|dd }tf|}t ||rt |||dk r|r|sd|_ d|_ | |j|d |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: rRFNrSTrQr_r`r^)rj)rr verifyrgrlenrbpoprrrrRZ AuxLogitsZload_state_dictZget_state_dict)rirjrfrZoriginal_aux_logitsr`modelr(r(r)rs.1       ) _ModelURLs) model_urlsZinception_v3_google_fbgemm)s`          "2 G