Yh($ ddlmZddlmZmZmZddlZddlmZmZddl m Z m Z ddl m Z mZddlmZd d lmZmZmZd d lmZd d lmZmZd d lmZmZmZmZmZddl m!Z!m"Z"gdZ#GddeZ$GddeZ%GddeZ&de'ede(deede)de)dede&fdZ*Gdd eZ+ed!"ed#d$f%dd&d'd(deee+efde)de)dede&f d)Z,dS)*)partial)AnyOptionalUnionN)nnTensor) DeQuantStub QuantStub)Conv2dNormActivationSqueezeExcitation)ImageClassification)register_modelWeights WeightsEnum)_IMAGENET_CATEGORIES)_ovewrite_named_paramhandle_legacy_interface)_mobilenet_v3_confInvertedResidualInvertedResidualConfigMobileNet_V3_Large_Weights MobileNetV3) _fuse_modules _replace_relu)QuantizableMobileNetV3#MobileNet_V3_Large_QuantizedWeightsmobilenet_v3_largecheZdZdZdededdffd ZdedefdZd d ee ddfd Z fd Z xZ S) QuantizableSqueezeExcitationrargskwargsreturnNctj|d<tj|i|tj|_dS)Nscale_activation)r Hardsigmoidsuper__init__ quantizedFloatFunctionalskip_mulselfr#r$ __class__s }/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/site-packages/torchvision/models/quantization/mobilenetv3.pyr*z%QuantizableSqueezeExcitation.__init__!sE%'^!"$)&))) 4466 inputc^|j|||SN)r-mul_scale)r/r3s r1forwardz$QuantizableSqueezeExcitation.forward&s&}  U!3!3U;;;r2is_qatc0t|ddg|ddS)Nfc1 activationTinplace)r)r/r9s r1 fuse_modelz'QuantizableSqueezeExcitation.fuse_model)s$dUL164HHHHHHr2c (|dd}t|dr||dkrtjdgtjdgtjdgtjtjdgtjtjdgtjdgd} | D]\} } || z} | |vr| || <t |||||||dS) Nversionqconfigrg?r)dtyper)z.scale_activation.activation_post_process.scalezFscale_activation.activation_post_process.activation_post_process.scalez3scale_activation.activation_post_process.zero_pointzKscale_activation.activation_post_process.activation_post_process.zero_pointz;scale_activation.activation_post_process.fake_quant_enabledz9scale_activation.activation_post_process.observer_enabled)gethasattrtorchtensorint32itemsr)_load_from_state_dict)r/ state_dictprefixlocal_metadatastrict missing_keysunexpected_keys error_msgsrAdefault_state_dictkvfull_keyr0s r1rJz2QuantizableSqueezeExcitation._load_from_state_dict,sC!$$Y55 4 # # -GaKKBG,PSuBUBUZ_ZfhkglZmZmGL|UVTW_d_jGkGkGk_d_kCu{```PU|]^\_O`O`MR\[\Z]M^M^ " " +0022 - -1!A::--+,Jx( %%            r2r5) __name__ __module__ __qualname___versionrr*rr8rboolr?rJ __classcell__r0s@r1r"r"sH7c7S7T777777 $P)EPPPP 4466 r2xc|jr.|j|||S||Sr5)use_res_connectraaddblockr/rbs r1r8z#QuantizableInvertedResidual.forwardYs?   !=$$Q 1 66 6::a== r2)rVrWrXrr*rr8r[r\s@r1r^r^Sst7c7S7T777777!!F!!!!!!!!r2r^cZeZdZdededdffd ZdedefdZd deeddfd Z xZ S) rr#r$r%Nctj|i|t|_t |_dS)zq MobileNet V3 main class Args: Inherits args from floating point MobileNetV3 N)r)r*r quantr dequantr.s r1r*zQuantizableMobileNetV3.__init__as; $)&)))[[ "}} r2rbc||}||}||}|Sr5)rj _forward_implrkrgs r1r8zQuantizableMobileNetV3.forwardls8 JJqMM   q ! ! LLOOr2r9cx|D]}t|turaddg}t|dkr6t|dtjur|dt|||dyt|tur| |dS)N01r r2Tr=) modulestyper lenrReLUappendrr"r?)r/r9mmodules_to_fuses r1r?z!QuantizableMobileNetV3.fuse_modelrs % %AAww...#&*q66Q;;4!::#8#8#**3///a&$GGGGGa888 V$$$ % %r2r5) rVrWrXrr*rr8rrZr?r[r\s@r1rr`s %c %S %T % % % % % %F %%$%4%%%%%%%%r2rinverted_residual_setting last_channelweightsprogressquantizer$r%c |Nt|dt|jdd|jvrt|d|jd|dd}t ||fdt i|}t ||re|dtj j ||_ tj j |d|*|||d |r:tj j |d||S) N num_classes categoriesbackendqnnpackrfT)r9r=)r| check_hash)rrtmetapoprr^rr?rFao quantizationget_default_qat_qconfigrB prepare_qatload_state_dictget_state_dictconverteval)ryrzr{r|r}r$rmodels r1_mobilenet_v3_modelr}sPfmSl9S5T5TUUU  $ $ !&)W\)5L M M MjjI..G "#>> g44hSW4XXYYY %%eT%::: Lr2cneZdZedeedddeddejdd d d id d dd Z e Z dS)rzUhttps://download.pytorch.org/models/quantized/mobilenet_v3_large_qnnpack-5bcacf28.pth) crop_sizeiS)rrrzUhttps://github.com/pytorch/vision/tree/main/references/classification#qat-mobilenetv3z ImageNet-1KgK7A@R@gxV@)zacc@1zacc@5g-?gҍ5@z These weights were produced by doing Quantization Aware Training (eager mode) on top of the unquantized weights listed below. ) num_paramsmin_sizerrrecipe unquantized_metrics_ops _file_size_docs)url transformsrN) rVrWrXrrrrr IMAGENET1K_V1IMAGENET1K_QNNPACK_V1DEFAULTr2r1rrs#G c7.#>>>!. m5C##      0$GGGr2rquantized_mobilenet_v3_large)name pretrainedc^|ddr tjn tjS)Nr}F)rDrrrr)r$s r1rs,zz*e,, : / E E+9r2)r{TF)r{r|r}c |rtnt|}tdi|\}}t |||||fi|S)a MobileNetV3 (Large) model from `Searching for MobileNetV3 `_. .. 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.MobileNet_V3_Large_QuantizedWeights` or :class:`~torchvision.models.MobileNet_V3_Large_Weights`, optional): The pretrained weights for the model. See :class:`~torchvision.models.quantization.MobileNet_V3_Large_QuantizedWeights` below for more details, and possible values. By default, no pre-trained weights are used. progress (bool): If True, displays a progress bar of the download to stderr. Default is True. quantize (bool): If True, return a quantized version of the model. Default is False. **kwargs: parameters passed to the ``torchvision.models.quantization.MobileNet_V3_Large_QuantizedWeights`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.quantization.MobileNet_V3_Large_QuantizedWeights :members: .. autoclass:: torchvision.models.MobileNet_V3_Large_Weights :members: :noindex: r )r )rrverifyrr)r{r|r}r$ryrzs r1r r sa^7?^22D^ffgnooG.@.`.`Y_.`.`+| 8,QY[c n ngm n nnr2)- functoolsrtypingrrrrFrrtorch.ao.quantizationr r ops.miscr r transforms._presetsr_apirrr_metar_utilsrr mobilenetv3rrrrrutilsrr__all__r"r^rlistintrZrrr rr2r1rs-'''''''''' 88888888????????6666667777777777((((((CCCCCCCC0///////   2 2 2 2 2 #42 2 2 j ! ! ! ! !"2 ! ! !%%%%%[%%%:!#$:;!!k "! !  !  !!!!!H$$$$$+$$$83444      ae 'o'o'o e?A[[\ ]'o'o 'o  'o  'o'o'o  54'o'o'or2