YhY* ddlZddlmZddlmZmZmZddlZddlm Z ddl m cm Z ddlm Z ddlmZddlmZmZddlmZd d lmZmZmZd d lmZd d lmZmZd dlm Z m!Z!m"Z"gdZ#Gddej$Z%Gddej&Z'Gddej(Z)Gddej*Z+Gddej,Z-Gddej.Z/Gddej0Z1Gddej2Z3Gd d!eZ4ed"#ed$d%f&dd'd(d)d*eee4efd+e5d,e5d-ed.e3f d/Z6dS)0N)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_v3cZeZdZdededdffd ZdedefdZd deeddfd Z xZ S) QuantizableBasicConv2dargskwargsreturnNcjtj|i|tj|_dSN)super__init__nnReLUreluselfrr __class__s {/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/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(r2s r* fuse_modelz!QuantizableBasicConv2d.fuse_model%s&d222FDIIIIIIr+r!) __name__ __module__ __qualname__rr#rr1rboolr5 __classcell__r)s@r*rrscSTF JJ$J4JJJJJJJJr+rc<eZdZdededdffd ZdedefdZxZS)QuantizableInceptionArrrNctj|dti|tj|_dSN conv_blockr"r#rr$ quantizedFloatFunctionalmyopr's r*r#zQuantizableInceptionA.__init__+>$L+ALVLLLL0022 r+r,cb||}|j|dSNr_forwardrDcatr(r,outputss r*r1zQuantizableInceptionA.forward/)--""y}}Wa(((r+r6r7r8rr#rr1r:r;s@r*r=r=)t3c3S3T333333))F))))))))r+r=c<eZdZdededdffd ZdedefdZxZS)QuantizableInceptionBrrrNctj|dti|tj|_dSr?rAr's r*r#zQuantizableInceptionB.__init__6rEr+r,cb||}|j|dSrGrHrKs r*r1zQuantizableInceptionB.forward:rMr+rNr;s@r*rQrQ4rOr+rQc<eZdZdededdffd ZdedefdZxZS)QuantizableInceptionCrrrNctj|dti|tj|_dSr?rAr's r*r#zQuantizableInceptionC.__init__ArEr+r,cb||}|j|dSrGrHrKs r*r1zQuantizableInceptionC.forwardErMr+rNr;s@r*rUrU?rOr+rUc<eZdZdededdffd ZdedefdZxZS)QuantizableInceptionDrrrNctj|dti|tj|_dSr?rAr's r*r#zQuantizableInceptionD.__init__LrEr+r,cb||}|j|dSrGrHrKs r*r1zQuantizableInceptionD.forwardPrMr+rNr;s@r*rYrYJrOr+rYcXeZdZdededdffd ZdedeefdZdedefdZxZ S) QuantizableInceptionErrrNctj|dti|tj|_tj|_tj|_dSr?) r"r#rr$rBrCmyop1myop2myop3r'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_rJbranch3x3dbl_1branch3x3dbl_2branch3x3dbl_3abranch3x3dbl_3br`F avg_pool2d branch_pool)r(r,rf branch3x3 branch3x3dblrprLs r*rIzQuantizableInceptionE._forward]sNN1%% $$Q'' &&y1143D3DY3O3OP JNN9a00 **1-- **<88   . .   . . z~~lA66 l1!AqIII &&{33 i{Cr+cb||}|j|dSrG)rIrarJrKs r*r1zQuantizableInceptionE.forwardrs)--""z~~gq)))r+) r6r7r8rr#rlistrIr1r:r;s@r*r]r]Us4c4S4T444444 &T&\***F********r+r]c,eZdZdededdffd ZxZS)QuantizableInceptionAuxrrrNcHtj|dti|dSr?)r"r#rr's r*r#z QuantizableInceptionAux.__init__ys+$L+ALVLLLLLr+)r6r7r8rr#r:r;s@r*rvrvws_McMSMTMMMMMMMMMMr+rvcZeZdZdededdffd ZdedefdZd dee ddfd Z xZ S) rrrrNc 2tj|dtttt t ttgi|tj j |_ tj j |_dS)Ninception_blocks)r"r#rr=rQrUrYr]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_inputrrIrtraining aux_logitsr{jit is_scriptingwarningswarnr eager_outputs)r(r,aux aux_defineds r*r1zQuantizableInception3.forwards  ! !! $ $ JJqMMq!!3 LLOOm7 9 ! ! # # . k ijjj#As++ +%%a-- -r+r2c|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)modulestyperr5)r(r2ms r*r5z QuantizableInception3.fuse_modelsJ % %AAww000 V$$$ % %r+r!) r6r7r8rr#rr r1rr9r5r: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) r6r7r8rrr 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,zz*e,, 4 ) > >%3r+)weightsTF)rprogressrrrrrrc |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*rrssh19R,,>RZZ[bccG **\599 F * * !&*;T B B BflD999fmSl9S5T5TUUU  $ $ !&)W\)5L M M MjjH--G ! + +F + +E%'ug&&&  #/ #$E "EO g44hSW4XXYYY # 3 #$E "EO Lr+)7r functoolsrtypingrrrr{torch.nnr$torch.nn.functional functionalrnrtorchvision.modelsrinception_moduletorchvision.models.inceptionr r transforms._presetsr _apirrr_metar_utilsrrutilsrrr__all__ BasicConv2dr InceptionAr= InceptionBrQ InceptionCrU InceptionDrY InceptionEr] InceptionAuxrv Inception3rrr9rrr+r*rs'''''''''' <<<<<<OOOOOOOO6666667777777777((((((CCCCCCCC??????????    J J J J J-9 J J J))))),7)))))))),7)))))))),7)))))))),7)))*****,7***DMMMMM.;MMM )%)%)%)%)%,7)%)%)%X#####K###8-...      UY CCC e9;OOP QCC C  C  CCC  /.CCCr+