a d9<@s ddlmZddlmZmZmZmZddlZddlm Z ddlm Z ddl m Z ddl mZdd lmZmZmZdd lmZdd lmZmZgd Ze ee d ddZGddde jZGddde jZeeeeeedddZdeddZ GdddeZ!GdddeZ"GdddeZ#Gd d!d!eZ$eed"e!j%fd#dd$d%ee!eeed&d'd(Z&eed"e"j%fd#dd$d%ee"eeed&d)d*Z'eed"e#j%fd#dd$d%ee#eeed&d+d,Z(eed"e$j%fd#dd$d%ee$eeed&d-d.Z)dS)/)partial)AnyCallableListOptionalN)Tensor)ImageClassification)_log_api_usage_once)register_modelWeights WeightsEnum)_IMAGENET_CATEGORIES)_ovewrite_named_paramhandle_legacy_interface) ShuffleNetV2ShuffleNet_V2_X0_5_WeightsShuffleNet_V2_X1_0_WeightsShuffleNet_V2_X1_5_WeightsShuffleNet_V2_X2_0_Weightsshufflenet_v2_x0_5shufflenet_v2_x1_0shufflenet_v2_x1_5shufflenet_v2_x2_0)xgroupsreturncCsP|\}}}}||}||||||}t|dd}||d||}|S)Nr r)sizeviewtorchZ transpose contiguous)rrZ batchsizeZ num_channelsheightwidthZchannels_per_groupr%h/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torchvision/models/shufflenetv2.pychannel_shuffles r'c sZeZdZeeeddfdd Zedeeeeeeejdd d Z e e d d d Z Z S)InvertedResidualN)inpoupstridercsNtd|krdks(ntd||_|d}|jdkrh||d>krhtd|d|d|d|jdkrt|j||d|jdd t|tj||ddd d d t|tj d d|_ n t|_ ttj|jdkr|n||ddd d d t|tj d d|j||d|jdd t|tj||ddd d d t|tj d d|_ dS)Nr zillegal stride valuerzInvalid combination of stride z, inp z and oup zB values. If stride == 1 then inp should be equal to oup // 2 << 1. kernel_sizer+paddingrF)r.r+r/biasTZinplace) super__init__ ValueErrorr+nn Sequentialdepthwise_conv BatchNorm2dConv2dReLUbranch1branch2)selfr)r*r+Zbranch_features __class__r%r&r3,sF      zInvertedResidual.__init__r rF)ior.r+r/r0rc Cstj|||||||dS)N)r0r)r5r9)r@rAr.r+r/r0r%r%r&r7VszInvertedResidual.depthwise_convrrcCsb|jdkr6|jddd\}}tj|||fdd}ntj||||fdd}t|d}|S)Nr r)Zdim)r+chunkr!catr<r;r')r=rx1Zx2outr%r%r&forward\s   zInvertedResidual.forward)r rF) __name__ __module__ __qualname__intr3 staticmethodboolr5r9r7rrG __classcell__r%r%r>r&r(+s* r(csbeZdZdefeeeeeedejfddfdd Z e e ddd Z e e dd d Z Z S) ri.N)stages_repeatsstages_out_channels num_classesinverted_residualrc sdtt|t|dkr&tdt|dkr:td||_d}|jd}ttj||ddddd t |tj d d |_ |}tj dddd |_ |||d ddD}t|||jddD]R\}} }|||dg} t| dD]} | |||dqt||tj| |}q|jd}ttj||ddddd t |tj d d |_t|||_dS)Nr,z2expected stages_repeats as list of 3 positive intsz7expected stages_out_channels as list of 5 positive intsrrr F)r0Tr1r-cSsg|]}d|qS)Zstager%).0r@r%r%r& z)ShuffleNetV2.__init__..)rr,r)r2r3r lenr4Z_stage_out_channelsr5r6r9r8r:conv1Z MaxPool2dmaxpoolziprangeappendsetattrconv5ZLinearfc) r=rOrPrQrRZinput_channelsZoutput_channelsZ stage_namesnameZrepeatsseqr@r>r%r&r3isB        zShuffleNetV2.__init__rBcCsX||}||}||}||}||}||}|ddg}||}|S)Nrr,)rYrZZstage2Zstage3Zstage4r_meanr`r=rr%r%r& _forward_impls       zShuffleNetV2._forward_implcCs ||S)N)rerdr%r%r&rGszShuffleNetV2.forward)rHrIrJr(rrKrr5Moduler3rrerGrNr%r%r>r&rhs 0 r)weightsprogressargskwargsrcOsJ|durt|dt|jdt|i|}|durF||j|d|S)NrQ categories)rh)rrXmetarZload_state_dictZget_state_dict)rgrhrirjmodelr%r%r& _shufflenetv2s rn)r r z2https://github.com/ericsun99/Shufflenet-v2-Pytorch)Zmin_sizerkrecipec @sDeZdZedeeddiedddddid d d d d ZeZdS)rzDhttps://download.pytorch.org/models/shufflenetv2_x0.5-f707e7126e.pth crop_sizei ImageNet-1Kg-FN@g9voT@zacc@1zacc@5g{Gz?gT㥛 @VThese weights were trained from scratch to reproduce closely the results of the paper. num_params_metrics_ops _file_size_docsurlZ transformsrlN rHrIrJr rr _COMMON_META IMAGENET1K_V1DEFAULTr%r%r%r&rs$ rc @sDeZdZedeeddiedddddid d d d d ZeZdS)rzBhttps://download.pytorch.org/models/shufflenetv2_x1-5666bf0f80.pthrprqi"rsgI +WQ@gNbX9V@rtg(\?gE!@rurvr|Nr~r%r%r%r&rs$ rc @sHeZdZedeedddieddddd d id d d ddZeZdS)rzBhttps://download.pytorch.org/models/shufflenetv2_x1_5-3c479a10.pthrprrZ resize_size+https://github.com/pytorch/vision/pull/5906iv5rsg9v?R@g/$V@rtgl?gw/+@ These weights were trained from scratch by using TorchVision's `new training recipe `_. rorwrxryrzr{r|Nr~r%r%r%r&rs& rc @sHeZdZedeedddieddddd d id d d ddZeZdS)rzBhttps://download.pytorch.org/models/shufflenetv2_x2_0-8be3c8ee.pthrprrriprsgQS@gMb@W@rtg-?g+n<@rrr|Nr~r%r%r%r&rs& rZ pretrained)rgT)rgrh)rgrhrjrcKs(t|}t||gdgdfi|S)a  Constructs a ShuffleNetV2 architecture with 0.5x output channels, as described in `ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Design `__. Args: weights (:class:`~torchvision.models.ShuffleNet_V2_X0_5_Weights`, optional): The pretrained weights to use. See :class:`~torchvision.models.ShuffleNet_V2_X0_5_Weights` 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. **kwargs: parameters passed to the ``torchvision.models.shufflenetv2.ShuffleNetV2`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.ShuffleNet_V2_X0_5_Weights :members: rWrW)0`)rverifyrnrgrhrjr%r%r&rs rcKs(t|}t||gdgdfi|S)a  Constructs a ShuffleNetV2 architecture with 1.0x output channels, as described in `ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Design `__. Args: weights (:class:`~torchvision.models.ShuffleNet_V2_X1_0_Weights`, optional): The pretrained weights to use. See :class:`~torchvision.models.ShuffleNet_V2_X1_0_Weights` 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. **kwargs: parameters passed to the ``torchvision.models.shufflenetv2.ShuffleNetV2`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.ShuffleNet_V2_X1_0_Weights :members: r)rtrir)rrrnrr%r%r&r>s rcKs(t|}t||gdgdfi|S)a  Constructs a ShuffleNetV2 architecture with 1.5x output channels, as described in `ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Design `__. Args: weights (:class:`~torchvision.models.ShuffleNet_V2_X1_5_Weights`, optional): The pretrained weights to use. See :class:`~torchvision.models.ShuffleNet_V2_X1_5_Weights` 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. **kwargs: parameters passed to the ``torchvision.models.shufflenetv2.ShuffleNetV2`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.ShuffleNet_V2_X1_5_Weights :members: r)ri`ir)rrrnrr%r%r&r]s rcKs(t|}t||gdgdfi|S)a  Constructs a ShuffleNetV2 architecture with 2.0x output channels, as described in `ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Design `__. Args: weights (:class:`~torchvision.models.ShuffleNet_V2_X2_0_Weights`, optional): The pretrained weights to use. See :class:`~torchvision.models.ShuffleNet_V2_X2_0_Weights` 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. **kwargs: parameters passed to the ``torchvision.models.shufflenetv2.ShuffleNetV2`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.ShuffleNet_V2_X2_0_Weights :members: r)riii)rrrnrr%r%r&r|s r)* functoolsrtypingrrrrr!Ztorch.nnr5rZtransforms._presetsr utilsr Z_apir r rZ_metar_utilsrr__all__rKr'rfr(rrMrnrrrrrrrrrrr%r%r%r&sn       =B