jP+ ddlZddlmZddlmZmZmZmZddlZddl m Z ddl m cm Z ddlmZddlmZddlmZmZddlmZd d lmZmZmZd d lmZd d lmZmZd dl m!Z!m"Z"m#Z#gdZ$Gddej%Z&Gddej'Z(Gddej)Z*Gddej+Z,Gddej-Z.Gddej/Z0Gddej1Z2Gddej3Z4Gd d!eZ5ed"#ed$d%f&dd'd(d)d*eee5efd+e6d,e6d-ed.e4f d/Z7dS)0N)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_v3cZeZdZdededdffd ZdedefdZd deeddfd Z xZ S) QuantizableBasicConv2dargskwargsreturnNcjtj|i|tj|_dSN)super__init__nnReLUreluselfrr __class__s ZC:\Projects\Historical\venv\Lib\site-packages\torchvision/models/quantization/inception.pyr$zQuantizableBasicConv2d.__init__s0$)&)))GII xc||}||}||}|Sr"convbnr')r)r-s r+forwardzQuantizableBasicConv2d.forwards4 IIaLL GGAJJ IIaLLr,is_qatc0t|gd|ddS)Nr/T)inplace)r)r)r3s r+ fuse_modelz!QuantizableBasicConv2d.fuse_model%s&d222FDIIIIIIr,r") __name__ __module__ __qualname__rr$rr2rboolr6 __classcell__r*s@r+rrscSTF JJ$J4JJJJJJJJr,rc<eZdZdededdffd ZdedefdZxZS)QuantizableInceptionArrr Nctj|dti|tj|_dSN conv_blockr#r$rr% quantizedFloatFunctionalmyopr(s r+r$zQuantizableInceptionA.__init__+>$L+ALVLLLL0022 r,r-cb||}|j|dSNr_forwardrEcatr)r-outputss r+r2zQuantizableInceptionA.forward/)--""y}}Wa(((r,r7r8r9rr$rr2r;r<s@r+r>r>)t3c3S3T333333))F))))))))r,r>c<eZdZdededdffd ZdedefdZxZS)QuantizableInceptionBrrr Nctj|dti|tj|_dSr@rBr(s r+r$zQuantizableInceptionB.__init__6rFr,r-cb||}|j|dSrHrIrLs r+r2zQuantizableInceptionB.forward:rNr,rOr<s@r+rRrR4rPr,rRc<eZdZdededdffd ZdedefdZxZS)QuantizableInceptionCrrr Nctj|dti|tj|_dSr@rBr(s r+r$zQuantizableInceptionC.__init__ArFr,r-cb||}|j|dSrHrIrLs r+r2zQuantizableInceptionC.forwardErNr,rOr<s@r+rVrV?rPr,rVc<eZdZdededdffd ZdedefdZxZS)QuantizableInceptionDrrr Nctj|dti|tj|_dSr@rBr(s r+r$zQuantizableInceptionD.__init__LrFr,r-cb||}|j|dSrHrIrLs r+r2zQuantizableInceptionD.forwardPrNr,rOr<s@r+rZrZJrPr,rZcXeZdZdededdffd ZdedeefdZdedefdZxZ S) QuantizableInceptionErrr Nctj|dti|tj|_tj|_tj|_dSr@) r#r$rr%rCrDmyop1myop2myop3r(s r+r$zQuantizableInceptionE.__init__Wsh$L+ALVLLL\1133 \1133 \1133 r,r-c(||}||}||||g}|j|d}||}||}||| |g}|j |d}tj |ddd}| |}||||g}|S)Nrr ) kernel_sizestridepadding) branch1x1 branch3x3_1 branch3x3_2a branch3x3_2br`rKbranch3x3dbl_1branch3x3dbl_2branch3x3dbl_3abranch3x3dbl_3braF avg_pool2d branch_pool)r)r-rg branch3x3 branch3x3dblrqrMs r+rJzQuantizableInceptionE._forward]sNN1%% $$Q'' &&y1143D3DY3O3OP JNN9a00 **1-- **<88   . .   . . z~~lA66 l1!AqIII &&{33 i{Cr,cb||}|j|dSrH)rJrbrKrLs r+r2zQuantizableInceptionE.forwardrs)--""z~~gq)))r,) r7r8r9rr$rrrJr2r;r<s@r+r^r^Us4c4S4T444444 &T&\***F********r,r^c,eZdZdededdffd ZxZS)QuantizableInceptionAuxrrr NcHtj|dti|dSr@)r#r$rr(s r+r$z QuantizableInceptionAux.__init__ys+$L+ALVLLLLLr,)r7r8r9rr$r;r<s@r+rvrvws_McMSMTMMMMMMMMMMr,rvcZeZdZdededdffd ZdedefdZd dee ddfd Z xZ S) rrrr Nc 2tj|dtttt t ttgi|tj j |_ tj j |_dS)Ninception_blocks)r#r$rr>rRrVrZr^rvtorchao quantization QuantStubquant DeQuantStubdequantr(s r+r$zQuantizableInception3.__init__~s  '%%%%%'  X*4466 x,88:: r,r-c||}||}||\}}||}|jo|j}t jr&|stj dt||S| ||S)NzIScripted QuantizableInception3 always returns QuantizableInception3 Tuple) _transform_inputrrJrtraining aux_logitsr{jit is_scriptingwarningswarnr eager_outputs)r)r-aux aux_defineds r+r2zQuantizableInception3.forwards  ! !! $ $ JJqMMq!!3 LLOOm7 9 ! ! # # . k ijjj#As++ +%%a-- -r,r3c|D]-}t|tur||.dS)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)modulestyperr6)r)r3ms r+r6z QuantizableInception3.fuse_modelsJ % %AAww000 V$$$ % %r,r") r7r8r9rr$rr r2rr:r6r;r<s@r+rr}s;c;S;T;;;;;;" . .$4 . . . . % %$ %4 % % % % % % % %r,rcpeZdZedeeddddeddejd d d d id ddd Z e Z dS)rzUhttps://download.pytorch.org/models/quantized/inception_v3_google_fbgemm-a2837893.pthi+iV) crop_size resize_sizeir)Krfbgemmzdhttps://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) r7r8r9rrr rr IMAGENET1K_V1IMAGENET1K_FBGEMM_V1DEFAULTr,r+rrs"7 c7.#3OOO" .|/=##      0#GGGr,rquantized_inception_v3)name pretrainedc^|ddr tjn tjS)NquantizeF)getrrr r)rs r+rs, ::j% ( (04II ! /r,)weightsTF)rprogressrrrrrr c |rtnt|}|dd}|td|vrt |ddt |ddt |dt |jdd|jvrt |d|jd|dd }td i|}t||rt|||N|r|sd|_ d|_ | ||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: rFNtransform_inputT num_classesrrr)r check_hashr)rr verifyrrlenrpoprrrr AuxLogitsload_state_dictget_state_dict)rrrroriginal_aux_logitsrmodels r+rrssd19R,,>RZZ[bccG **\599 F * * !&*;T B B BflD999fmSl9S5T5TUUU  $ $ !&)W\)5L M M MjjH--G ! + +F + +E%'ug&&&  #/ #$E "EO g44hSW4XXYYY # 3 #$E "EO Lr,)8r functoolsrtypingrrrrr{torch.nnr%torch.nn.functional functionalrortorchvision.modelsr inception_moduletorchvision.models.inceptionr r transforms._presetsr _apirrr_metar_utilsrrutilsrrr__all__ BasicConv2dr InceptionAr> InceptionBrR InceptionCrV InceptionDrZ InceptionEr^ InceptionAuxrv Inception3rrr:rrr,r+rs------------ <<<<<<OOOOOOOO6666667777777777((((((CCCCCCCC??????????    J J J J J-9 J J J))))),7)))))))),7)))))))),7)))))))),7)))*****,7***DMMMMM.;MMM )%)%)%)%)%,7)%)%)%X#####K###8-... 0 0 UY CCC e9;OOP QCC C  C  CCC/.CCCr,