U hA @sddlmZddlmZmZmZmZddlZddlm Z ddlm Z ddl m Z ddl mZdd lmZmZmZdd lmZdd lmZmZdd l mZmZmZmZd dlmZmZmZdddddddddg Z Gddde j!Z"Gddde j#Z$ee%ee%eee&e&ee$dddZ'dedd d!d"Z(Gd#ddeZ)Gd$ddeZ*Gd%ddeZ+Gd&ddeZ,ed'd(ed)d*d+fd,dd-d.d/eee)efe&e&ee$d0d1dZ-ed2d(ed)d3d+fd,dd-d.d/eee*efe&e&ee$d0d4dZ.ed5d(ed)d6d+fd,dd-d.d/eee+efe&e&ee$d0d7dZ/ed8d(ed)d9d+fd,dd-d.d/eee,efe&e&ee$d0d:dZ0dS);)partial)AnyListOptionalUnionN)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_modelQuantizableShuffleNetV2#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_0cs6eZdZeeddfdd ZeedddZZS)QuantizableInvertedResidualNargskwargsreturncstj||tj|_dSN)super__init__nnZ quantizedZFloatFunctionalcatselfr%r& __class__^/var/www/html/venv/lib/python3.8/site-packages/torchvision/models/quantization/shufflenetv2.pyr*$sz$QuantizableInvertedResidual.__init__xr'cCsh|jdkr8|jddd\}}|jj|||gdd}n |jj||||gdd}t|d}|S)Nrr )dim)Zstridechunkr,branch2branch1rZchannel_shuffle)r.r4x1Zx2outr1r1r2forward(s    z#QuantizableInvertedResidual.forward)__name__ __module__ __qualname__rr*rr; __classcell__r1r1r/r2r##sr#csLeZdZeeddfdd ZeedddZd eeddd d Z Z S) rNr$cs6tj|dti|tjj|_tjj|_ dS)NZinverted_residual) r)r*r#torchZaoZ quantizationZ QuantStubquantZ DeQuantStubdequantr-r/r1r2r*6sz QuantizableShuffleNetV2.__init__r3cCs"||}||}||}|Sr()rAZ _forward_implrB)r.r4r1r1r2r;;s   zQuantizableShuffleNetV2.forward)is_qatr'cCs|jD]0\}}|dkr |dk r t|dddgg|ddq |D]l}t|tkrDt|jjdkrt|jddgdd d gg|ddt|jdddgd d gd d d gg|ddqDdS)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 )Zconv1Zconv5N012T)Zinplacer34567) Z_modulesitemsrmodulestyper#lenr8r7)r.rCnamemr1r1r2 fuse_modelAs    z"QuantizableShuffleNetV2.fuse_model)N) r<r=r>rr*rr;rboolrRr?r1r1r/r2r4s)stages_repeatsstages_out_channelsweightsprogressquantizer&r'cKs|dk r:t|dt|jdd|jkr:t|d|jd|dd}t||f|}t||rjt|||dk r||j|dd|S)NZ num_classes categoriesbackendfbgemmT)rWZ check_hash) rrOmetapoprrrZload_state_dictZget_state_dict)rTrUrVrWrXr&rZmodelr1r1r2 _shufflenetv2Zs    r_)rrr[zdhttps://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_sizerYrZrecipeZ_docsc @sBeZdZedeeddedejddddid 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 transformsr\N) r<r=r>r rr _COMMON_METAr IMAGENET1K_V1IMAGENET1K_FBGEMM_V1DEFAULTr1r1r1r2rs  c @sBeZdZedeeddedejddddid d d d Z e Z d S)rzQhttps://download.pytorch.org/models/quantized/shufflenetv2_x1_fbgemm-1e62bb32.pthrarbi"rdgףp= Q@gh|?U@reg(\?gy&1@rfrlN) r<r=r>r rr rnrrorprqr1r1r1r2rs  c @sFeZdZedeedddeddejddd d id d d dZ e Z dS)rzShttps://download.pytorch.org/models/quantized/shufflenetv2_x1_5_fbgemm-d7401f05.pthrarcZ resize_size+https://github.com/pytorch/vision/pull/5906iv5rdgSR@g̬V@regl?gK7A` @r`rgrhrirjrkrlN) r<r=r>r rr rnrrorprqr1r1r1r2rs" c @sFeZdZedeedddeddejddd d id d d dZ e Z dS)rzShttps://download.pytorch.org/models/quantized/shufflenetv2_x2_0_fbgemm-5cac526c.pthrarrrsrtiprdg-R@gZd;W@reg-?g|?5@rurlN) r<r=r>r rr rnrrorprqr1r1r1r2rs" Zquantized_shufflenet_v2_x0_5)rPZ pretrainedcCs|ddrtjStjSNrXF)getrrprror&r1r1r2s ry)rVTFrVrWrX)rVrWrXr&r'cKs<|rtnt|}tdddgdddddgf|||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: 0`rz)rrverifyr_rVrWrXr&r1r1r2rs0 Zquantized_shufflenet_v2_x1_0cCs|ddrtjStjSrv)rwrrprrorxr1r1r2rys cKs<|rtnt|}tdddgdddddgf|||d|S) 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{r|r}trrirrz)rrrr_rr1r1r2r s0 Zquantized_shufflenet_v2_x1_5cCs|ddrtjStjSrv)rwrrprrorxr1r1r2ryFs cKs<|rtnt|}tdddgdddddgf|||d|S) 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{r|r}i`irrz)rrrr_rr1r1r2r!Bs0 Zquantized_shufflenet_v2_x2_0cCs|ddrtjStjSrv)rwrrprrorxr1r1r2ry|s cKs<|rtnt|}tdddgdddddgf|||d|S) 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{r|r}iiirz)rrrr_rr1r1r2r"xs0 )1 functoolsrtypingrrrrr@Ztorch.nnr+rZtorchvision.modelsrZtransforms._presetsr Z_apir r r_metarZ_utilsrrrrrrutilsrrr__all__ZInvertedResidualr#Z ShuffleNetV2rintrSr_rnrrrrrr r!r"r1r1r1r2s       '   - - -