:jY* ddlZddlmZddlmZmZmZddlZddlm Z ddl m cm Z ddlm Z ddlmZddlmZmZddlmZd d lmZmZmZd d lmZd d lmZmZd dlm Z m!Z!m"Z"gdZ#GddejHZ%GddejLZ'GddejPZ)GddejTZ+GddejXZ-Gddej\Z/Gddej`Z1GddejdZ3Gd d!eZ4ed"#ed$d%f&dd'd(d)d*eee4efd+e5d,e5d-ed.e3f d/Z6y)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_v3cTeZdZdededdffd ZdedefdZd deeddfd Z xZ S) QuantizableBasicConv2dargskwargsreturnNcVt||i|tj|_yN)super__init__nnReLUreluselfrr __class__s ~/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/torchvision/models/quantization/inception.pyr#zQuantizableBasicConv2d.__init__s" $)&)GGI xcl|j|}|j|}|j|}|Sr!convbnr&)r(r,s r*forwardzQuantizableBasicConv2d.forwards. IIaL GGAJ IIaLr+is_qatc&t|gd|dy)Nr.T)inplace)r)r(r2s r* fuse_modelz!QuantizableBasicConv2d.fuse_model%sd2FDIr+r!) __name__ __module__ __qualname__rr#rr1rboolr5 __classcell__r)s@r*rrsJcSTF J$J4Jr+rc<eZdZdededdffd ZdedefdZxZS)QuantizableInceptionArrrNcvt||dti|tjj |_yN conv_blockr"r#rr$ quantizedFloatFunctionalmyopr's r*r#zQuantizableInceptionA.__init__+/ $L+ALVLLL002 r+r,c\|j|}|jj|dSNr_forwardrDcatr(r,outputss r*r1zQuantizableInceptionA.forward/%--"yy}}Wa((r+r6r7r8rr#rr1r:r;s@r*r=r=)03c3S3T3))F)r+r=c<eZdZdededdffd ZdedefdZxZS)QuantizableInceptionBrrrNcvt||dti|tjj |_yr?rAr's r*r#zQuantizableInceptionB.__init__6rEr+r,c\|j|}|jj|dSrGrHrKs r*r1zQuantizableInceptionB.forward:rMr+rNr;s@r*rQrQ4rOr+rQc<eZdZdededdffd ZdedefdZxZS)QuantizableInceptionCrrrNcvt||dti|tjj |_yr?rAr's r*r#zQuantizableInceptionC.__init__ArEr+r,c\|j|}|jj|dSrGrHrKs r*r1zQuantizableInceptionC.forwardErMr+rNr;s@r*rUrU?rOr+rUc<eZdZdededdffd ZdedefdZxZS)QuantizableInceptionDrrrNcvt||dti|tjj |_yr?rAr's r*r#zQuantizableInceptionD.__init__LrEr+r,c\|j|}|jj|dSrGrHrKs r*r1zQuantizableInceptionD.forwardPrMr+rNr;s@r*rYrYJrOr+rYcReZdZdededdffd ZdedeefdZdedefdZxZ S) QuantizableInceptionErrrNct||dti|tjj |_tjj |_tjj |_yr?) r"r#rr$rBrCmyop1myop2myop3r's r*r#zQuantizableInceptionE.__init__WsW $L+ALVL\\113 \\113 \\113 r+r,c|j|}|j|}|j||j|g}|jj |d}|j |}|j|}|j||j|g}|jj |d}tj|ddd}|j|}||||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' &&y143D3DY3OP JJNN9a0 **1- **<8   .   . zz~~lA6 ll1!AqI &&{3 i{Cr+c\|j|}|jj|dSrG)rIrarJrKs r*r1zQuantizableInceptionE.forwardrs%--"zz~~gq))r+) r6r7r8rr#rlistrIr1r:r;s@r*r]r]UsG4c4S4T4 &T&\***F*r+r]c,eZdZdededdffd ZxZS)QuantizableInceptionAuxrrrNc0t||dti|yr?)r"r#rr's r*r#z QuantizableInceptionAux.__init__ys $L+ALVLr+)r6r7r8rr#r:r;s@r*rvrvws'McMSMTMMr+rvcTeZdZdededdffd ZdedefdZd dee ddfd Z xZ S) rrrrNc "t||dtttt t ttgi|tjjj|_ tjjj|_y)Ninception_blocks)r"r#rr=rQrUrYr]rvtorchao quantization QuantStubquant DeQuantStubdequantr's r*r#zQuantizableInception3.__init__~st   '%%%%%'  XX**446 xx,,88: r+r,cj|j|}|j|}|j|\}}|j|}|jxr |j }t jjr#|stjdt||S|j||S)NzIScripted QuantizableInception3 always returns QuantizableInception3 Tuple) _transform_inputrrIrtraining aux_logitsr{jit is_scriptingwarningswarnr eager_outputs)r(r,aux aux_defineds r*r1zQuantizableInception3.forwards  ! !! $ JJqMq!3 LLOmm7 99 ! ! # ij#As+ +%%a- -r+r2ct|jD]%}t|tus|j|'y)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_models2 %AAw00 V$ %r+r!) r6r7r8rr#rr r1rr9r5r:r;s@r*rr}sH;c;S;T;" . .$4 . %$ %4 %r+rcleZdZedeeddddeddejd d d d id ddd Z e Z y)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*rrsa" c.#3O" .|/==##    0#Gr+rquantized_inception_v3)name pretrainedcf|jddrtjStjS)NquantizeF)getrrr r)rs r*rs0zz*e, * > >&33r+)weightsTF)rprogressrrrrrrc 6|rtntj|}|jdd}|id|vr t |ddt |ddt |dt |j dd|j vrt |d|j d|jdd }td i|}t||r t|||F|r|sd|_ d|_ |j|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: rFNtransform_inputT num_classesrrr)r check_hashr)rr verifyrrlenrpoprrrr AuxLogitsload_state_dictget_state_dict)rrrroriginal_aux_logitsrmodels r*rrsh19,>RZZ[bcG **\59 F * !&*;T BflD9fmSl9S5TU  $ !&)W\\)5L MjjH-G ! +F +E%ug& /$E "EO g44hSW4XY 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*rsu'' <O677(C??  J-99 J),77)),77)),77)),77)*,77*DM.;;M )%,77)%X#K#8-.   UY C e9;OOP QCC C  C  C /Cr+