o )i @spddlZddlmZddlmZmZmZddlZddlm Z ddlm Z ddlm Z ddl mZdd lmZmZmZdd lmZdd lmZmZdd lmZmZmZmZmZmZd dlm Z m!Z!m"Z"gdZ#GdddeZ$GdddeZ%GdddeZ&GdddeZ'GdddeZ(eddedddfddd d!d"d#eee(efd$e)d%e)d&ed'e'f d(d)ZdS)*N)partial)AnyOptionalUnion)Tensor) functional)ImageClassification)register_modelWeights WeightsEnum)_IMAGENET_CATEGORIES)_ovewrite_named_paramhandle_legacy_interface) BasicConv2d GoogLeNetGoogLeNet_WeightsGoogLeNetOutputs Inception InceptionAux) _fuse_modules _replace_reluquantize_model)QuantizableGoogLeNetGoogLeNet_QuantizedWeights googlenetcsTeZdZdededdffdd Zdedefdd Zd d eeddfd d Z Z S)QuantizableBasicConv2dargskwargsreturnNcs tj|i|t|_dSN)super__init__nnReLUreluselfrr  __class__x/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/torchvision/models/quantization/googlenet.pyr$szQuantizableBasicConv2d.__init__xcCs"||}||}||}|Sr"convZbnr'r)r.r,r,r-forwards   zQuantizableBasicConv2d.forwardis_qatcCst|gd|dddS)Nr/T)Zinplace)r)r)r3r,r,r- fuse_model$sz!QuantizableBasicConv2d.fuse_modelr") __name__ __module__ __qualname__rr$rr2rboolr4 __classcell__r,r,r*r-rs rc<eZdZdededdffdd Zdedefdd ZZS) QuantizableInceptionrr r!Ncs&tj|dti|tj|_dSNZ conv_block)r#r$rr%Z quantizedZFloatFunctionalcatr(r*r,r-r$)szQuantizableInception.__init__r.cCs||}|j|dS)Nr)_forwardr=)r)r.Zoutputsr,r,r-r2-s zQuantizableInception.forwardr5r6r7rr$rr2r9r,r,r*r-r;(sr;cr:) QuantizableInceptionAuxrr r!Ncs$tj|dti|t|_dSr<)r#r$rr%r&r'r(r*r,r-r$4sz QuantizableInceptionAux.__init__r.cCsJt|d}||}t|d}|||}||}||}|S)N)rAr) FZadaptive_avg_pool2dr0torchflattenr'Zfc1ZdropoutZfc2r1r,r,r-r28s     zQuantizableInceptionAux.forwardr?r,r,r*r-r@2sr@csTeZdZdededdffdd Zdedefdd Zd d ee ddfd d Z Z S)rrr r!Ncs<tj|dtttgi|tjj|_ tjj |_ dS)Nblocks) r#r$rr;r@rCZaoZ quantizationZ QuantStubquantZ DeQuantStubdequantr(r*r,r-r$LszQuantizableGoogLeNet.__init__r.cCsl||}||}||\}}}||}|jo|j}tjr/|s)t dt |||S| |||S)NzCScripted QuantizableGoogleNet always returns GoogleNetOutputs Tuple) Z_transform_inputrFr>rGZtraining aux_logitsrCZjitZ is_scriptingwarningswarnrZ eager_outputs)r)r.aux1aux2Z aux_definedr,r,r-r2Ss       zQuantizableGoogLeNet.forwardr3cCs(|D] }t|tur||qdS)aFuse conv/bn/relu modules in googlenet 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)modulestyperr4)r)r3mr,r,r-r4`s   zQuantizableGoogLeNet.fuse_modelr") r5r6r7rr$rrr2rr8r4r9r,r,r*r-rJs rc@sHeZdZedeeddddeddejdd d d id d dd dZ e Z dS)rzKhttps://download.pytorch.org/models/quantized/googlenet_fbgemm-c81f6644.pth)Z crop_sizeie)rQfbgemmzdhttps://github.com/pytorch/vision/tree/main/references/classification#post-training-quantized-modelsz ImageNet-1Kg/tQ@g`"YV@)zacc@1zacc@5g+?g#~j<)@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) r5r6r7r rr rr IMAGENET1K_V1IMAGENET1K_FBGEMM_V1DEFAULTr,r,r,r-rms* rZquantized_googlenet)nameZ pretrainedcCs|ddr tjStjS)NquantizeF)getrrYrrX)r r,r,r-s r^)weightsTF)r_progressr\r_r`r\r r!cKs|rtnt|}|dd}|durBd|vrt|ddt|ddt|ddt|dt|jdd |jvrBt|d |jd |d d }td i|}t ||rZt |||durz| |j |dd |sud|_ d|_d|_|Std |S)aGoogLeNet (Inception v1) model architecture from `Going Deeper with Convolutions `__. .. 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.GoogLeNet_QuantizedWeights` or :class:`~torchvision.models.GoogLeNet_Weights`, optional): The pretrained weights for the model. See :class:`~torchvision.models.quantization.GoogLeNet_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.QuantizableGoogLeNet`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.quantization.GoogLeNet_QuantizedWeights :members: .. autoclass:: torchvision.models.GoogLeNet_Weights :members: :noindex: rHFNZtransform_inputTZ init_weightsZ num_classesrSrTrR)r`Z check_hashz`auxiliary heads in the pretrained googlenet model are NOT pretrained, so make sure to train themr,)rrverifyr]rlenrWpoprrrZload_state_dictZget_state_dictrHrKrLrIrJ)r_r`r\r Zoriginal_aux_logitsrTmodelr,r,r-rs4,       r)*rI functoolsrtypingrrrrCZtorch.nnr%rrrBZtransforms._presetsr Z_apir r r _metar_utilsrrrrrrrrrutilsrrr__all__rr;r@rrr8r,r,r,r-sN        #