o hnB@sddlmZddlmZmZmZddlZddlmZddlm Z ddl m Z ddl m Z dd lmZmZmZdd lmZdd lmZmZdd l mZmZmZmZd dlmZmZmZgdZGddde j Z!Gddde j"Z#de$e%de$e%deede&de&dede#fddZ'deddd d!Z(Gd"d#d#eZ)Gd$d%d%eZ*Gd&d'd'eZ+Gd(d)d)eZ,ed*d+ed,d-d.fd/dd0d1d2deee)efde&de&dede#f d3d4Z-ed5d+ed,d6d.fd/dd0d1d2deee*efde&de&dede#f d7d8Z.ed9d+ed,d:d.fd/dd0d1d2deee+efde&de&dede#f d;d<Z/ed=d+ed,d>d.fd/dd0d1d2deee,efde&de&dede#f d?d@Z0dS)A)partial)AnyOptionalUnionN)Tensor) shufflenetv2)ImageClassification)register_modelWeights WeightsEnum)_IMAGENET_CATEGORIES)_ovewrite_named_paramhandle_legacy_interface)ShuffleNet_V2_X0_5_WeightsShuffleNet_V2_X1_0_WeightsShuffleNet_V2_X1_5_WeightsShuffleNet_V2_X2_0_Weights) _fuse_modules _replace_reluquantize_model) QuantizableShuffleNetV2#ShuffleNet_V2_X0_5_QuantizedWeights#ShuffleNet_V2_X1_0_QuantizedWeights#ShuffleNet_V2_X1_5_QuantizedWeights#ShuffleNet_V2_X2_0_QuantizedWeightsshufflenet_v2_x0_5shufflenet_v2_x1_0shufflenet_v2_x1_5shufflenet_v2_x2_0cs<eZdZdededdffdd Zdedefdd ZZS) QuantizableInvertedResidualargskwargsreturnNcs"tj|i|tj|_dSN)super__init__nnZ quantizedZFloatFunctionalcatselfr#r$ __class__r/home/www/facesmatcher.com/frenv_anti/lib/python3.10/site-packages/torchvision/models/quantization/shufflenetv2.pyr($sz$QuantizableInvertedResidual.__init__xcCsh|jdkr|jddd\}}|jj|||gdd}n|jj||||gdd}t|d}|S)Nrr )dim)Zstridechunkr*branch2branch1rZchannel_shuffle)r,r1x1Zx2outr/r/r0forward(s   z#QuantizableInvertedResidual.forward)__name__ __module__ __qualname__rr(rr8 __classcell__r/r/r-r0r"#sr"csTeZdZdededdffdd Zdedefdd Zd d eeddfd d Z Z S)rr#r$r%Ncs6tj|dti|tjj|_tjj|_ dS)NZinverted_residual) r'r(r"torchZaoZ quantizationZ QuantStubquantZ DeQuantStubdequantr+r-r/r0r(6sz QuantizableShuffleNetV2.__init__r1cCs"||}||}||}|Sr&)r>Z _forward_implr?)r,r1r/r/r0r8;s   zQuantizableShuffleNetV2.forwardis_qatcCs|jD]\}}|dvr|durt|gdg|ddq|D]3}t|turTt|jjdkrBt|jddggd g|ddt|jgdd d ggd g|ddq!dS) aBFuse conv/bn/relu modules in shufflenetv2 model Fuse conv+bn+relu/ conv+relu/conv+bn modules to prepare for quantization. Model is modified in place. .. note:: Note that this operation does not change numerics and the model after modification is in floating point )Zconv1Zconv5N)012T)ZinplacerrArB)rC34rDrE)567) Z_modulesitemsrmodulestyper"lenr5r4)r,r@namemr/r/r0 fuse_modelAs    z"QuantizableShuffleNetV2.fuse_modelr&) r9r:r;rr(rr8rboolrOr<r/r/r-r0r4s rstages_repeatsstages_out_channelsweightsprogressquantizer$r%cKs|durt|dt|jdd|jvrt|d|jd|dd}t||fi|}t||r7t|||durE||j|dd|S)NZ num_classes categoriesbackendfbgemmT)rTZ check_hash) rrLmetapoprrrZload_state_dictZget_state_dict)rQrRrSrTrUr$rWmodelr/r/r0 _shufflenetv2Zs    r\)rrrXzdhttps://github.com/pytorch/vision/tree/main/references/classification#post-training-quantized-modelsz These weights were produced by doing Post Training Quantization (eager mode) on top of the unquantized weights listed below. )Zmin_sizerVrWrecipeZ_docsc @FeZdZedeeddiedejddddid d d d Z e Z d S)rzShttps://download.pytorch.org/models/quantized/shufflenetv2_x0.5_fbgemm-00845098.pth crop_sizei ImageNet-1Kg#~jL@gRS@zacc@1zacc@5g{Gz?gjt? num_params unquantized_metrics_ops _file_sizeurlZ transformsrYN) r9r:r;r rr _COMMON_METAr IMAGENET1K_V1IMAGENET1K_FBGEMM_V1DEFAULTr/r/r/r0r& rc @r^)rzQhttps://download.pytorch.org/models/quantized/shufflenetv2_x1_fbgemm-1e62bb32.pthr_r`i"rbgףp= Q@gh|?U@rcg(\?gy&1@rdrjN) r9r:r;r rr rlrrmrnror/r/r/r0rrprc @JeZdZedeedddieddejddd d id d d dZ e Z dS)rzShttps://download.pytorch.org/models/quantized/shufflenetv2_x1_5_fbgemm-d7401f05.pthr_raZ resize_size+https://github.com/pytorch/vision/pull/5906iv5rbgSR@g̬V@rcgl?gK7A` @r]rerfrgrhrirjN) r9r:r;r rr rlrrmrnror/r/r/r0r( rc @rq)rzShttps://download.pytorch.org/models/quantized/shufflenetv2_x2_0_fbgemm-5cac526c.pthr_rrrsrtiprbg-R@gZd;W@rcg-?g|?5@rurjN) r9r:r;r rr rlrrmrnror/r/r/r0rrvrZquantized_shufflenet_v2_x0_5)rMZ pretrainedcC|ddr tjStjSNrUF)getrrnrrmr$r/r/r0 r{)rSTFrSrTrUcK4|rtnt|}tgdgdf|||d|S)aQ Constructs a ShuffleNetV2 with 0.5x output channels, as described in `ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Design `__. .. 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.ShuffleNet_V2_X0_5_QuantizedWeights` or :class:`~torchvision.models.ShuffleNet_V2_X0_5_Weights`, optional): The pretrained weights for the model. See :class:`~torchvision.models.quantization.ShuffleNet_V2_X0_5_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.ShuffleNet_V2_X0_5_QuantizedWeights`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.quantization.ShuffleNet_V2_X0_5_QuantizedWeights :members: .. autoclass:: torchvision.models.ShuffleNet_V2_X0_5_Weights :members: :noindex: r)0`r})rrverifyr\rSrTrUr$r/r/r0r2 rZquantized_shufflenet_v2_x1_0cCrwrx)ryrrnrrmrzr/r/r0r{r|cKr~)aQ Constructs a ShuffleNetV2 with 1.0x output channels, as described in `ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Design `__. .. 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.ShuffleNet_V2_X1_0_QuantizedWeights` or :class:`~torchvision.models.ShuffleNet_V2_X1_0_Weights`, optional): The pretrained weights for the model. See :class:`~torchvision.models.quantization.ShuffleNet_V2_X1_0_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.ShuffleNet_V2_X1_0_QuantizedWeights`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.quantization.ShuffleNet_V2_X1_0_QuantizedWeights :members: .. autoclass:: torchvision.models.ShuffleNet_V2_X1_0_Weights :members: :noindex: r)rtrrirr})rrrr\rr/r/r0rrrZquantized_shufflenet_v2_x1_5cCrwrx)ryrrnrrmrzr/r/r0r{Jr|cKr~)aQ Constructs a ShuffleNetV2 with 1.5x output channels, as described in `ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Design `__. .. 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.ShuffleNet_V2_X1_5_QuantizedWeights` or :class:`~torchvision.models.ShuffleNet_V2_X1_5_Weights`, optional): The pretrained weights for the model. See :class:`~torchvision.models.quantization.ShuffleNet_V2_X1_5_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.ShuffleNet_V2_X1_5_QuantizedWeights`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.quantization.ShuffleNet_V2_X1_5_QuantizedWeights :members: .. autoclass:: torchvision.models.ShuffleNet_V2_X1_5_Weights :members: :noindex: r)ri`irr})rrrr\rr/r/r0r Frr Zquantized_shufflenet_v2_x2_0cCrwrx)ryrrnrrmrzr/r/r0r{r|cKr~)aQ Constructs a ShuffleNetV2 with 2.0x output channels, as described in `ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Design `__. .. 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.ShuffleNet_V2_X2_0_QuantizedWeights` or :class:`~torchvision.models.ShuffleNet_V2_X2_0_Weights`, optional): The pretrained weights for the model. See :class:`~torchvision.models.quantization.ShuffleNet_V2_X2_0_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.ShuffleNet_V2_X2_0_QuantizedWeights`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.quantization.ShuffleNet_V2_X2_0_QuantizedWeights :members: .. autoclass:: torchvision.models.ShuffleNet_V2_X2_0_Weights :members: :noindex: r)riiir})rrrr\rr/r/r0r!~rr!)1 functoolsrtypingrrrr=Ztorch.nnr)rZtorchvision.modelsrZtransforms._presetsr Z_apir r r _metarZ_utilsrrrrrrutilsrrr__all__ZInvertedResidualr"Z ShuffleNetV2rlistintrPr\rlrrrrrrr r!r/r/r/r0s       &   - - -