o )iK @sUddlmZddlmZmZmZmZmZmZddl Z ddl m Z ddl m Z ddlmZddlmZmZmZdd lmZdd lmZmZgd ZGd d d e jZdGdeeeefdede jfddZ gdgdgdgddZ!eeeeeeffe"d<dededeedededef ddZ#ded d!d"Z$Gd#d$d$eZ%Gd%d&d&eZ&Gd'd(d(eZ'Gd)d*d*eZ(Gd+d,d,eZ)Gd-d.d.eZ*Gd/d0d0eZ+Gd1d2d2eZ,eed3e%j-fd4dd5d6dee%dededefd7d8Z.eed3e&j-fd4dd5d6dee&dededefd9d:Z/eed3e'j-fd4dd5d6dee'dededefd;d<Z0eed3e(j-fd4dd5d6dee(dededefd=d>Z1eed3e)j-fd4dd5d6dee)dededefd?d@Z2eed3e*j-fd4dd5d6dee*dededefdAdBZ3eed3e+j-fd4dd5d6dee+dededefdCdDZ4eed3e,j-fd4dd5d6dee,dededefdEdFZ5dS)H)partial)AnycastDictListOptionalUnionN)ImageClassification)_log_api_usage_once)register_modelWeights WeightsEnum)_IMAGENET_CATEGORIES)_ovewrite_named_paramhandle_legacy_interface)VGG VGG11_WeightsVGG11_BN_Weights VGG13_WeightsVGG13_BN_Weights VGG16_WeightsVGG16_BN_Weights VGG19_WeightsVGG19_BN_Weightsvgg11vgg11_bnvgg13vgg13_bnvgg16vgg16_bnvgg19vgg19_bnc sNeZdZ ddejdedededd f fd d Zd e j de j fd dZ Z S)rT?features num_classes init_weightsdropoutreturnNc s"tt|||_td|_ttddt dtj |dtddt dtj |dtd||_ |r| D]P}t |tjr^tjj|jddd|jdur]tj|jd q>t |tjrutj|jd tj|jd q>t |tjrtj|jd d tj|jd q>dSdS) N)r+ibiT)pZfan_outZrelu)modeZ nonlinearityrr g{Gz?)super__init__r r&nnZAdaptiveAvgPool2davgpool SequentialZLinearReLUZDropout classifiermodules isinstanceConv2dinitZkaiming_normal_weightZbiasZ constant_ BatchNorm2dZnormal_)selfr&r'r(r)m __class__e/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/torchvision/models/vgg.pyr/$s:            z VGG.__init__xcCs.||}||}t|d}||}|S)Nr )r&r1torchflattenr4)r;rAr?r?r@forwardAs    z VGG.forward)r$Tr%) __name__ __module__ __qualname__r0Moduleintboolfloatr/rBZTensorrD __classcell__r?r?r=r@r#srFcfg batch_normr*cCsg}d}|D]<}|dkr|tjdddg7}qtt|}tj||ddd}|r6||t|tjddg7}n ||tjddg7}|}qtj|S) NMr ) kernel_sizeZstrider )rQpaddingT)Zinplace)r0Z MaxPool2drrIr7r:r3r2)rMrNZlayersZ in_channelsvZconv2dr?r?r@ make_layersIs  rT) @rPrPrWrPrXrPrXrXrP)rUrUrPrVrVrPrWrWrPrXrXrPrXrXrP)rUrUrPrVrVrPrWrWrWrPrXrXrXrPrXrXrXrP)rUrUrPrVrVrPrWrWrWrWrPrXrXrXrXrPrXrXrXrXrP)ABDEcfgsweightsprogresskwargscKsp|durd|d<|jddurt|dt|jdttt||dfi|}|dur6||j|dd|S)NFr( categoriesr')rNT)r_Z check_hash)metarlenrrTr]Zload_state_dictZget_state_dict)rMrNr^r_r`modelr?r?r@_vggbsre) rfzUhttps://github.com/pytorch/vision/tree/main/references/classification#alexnet-and-vggzNThese weights were trained from scratch by using a simplified training recipe.)Zmin_sizerarecipe_docsc @BeZdZedeeddiedddddid d d d ZeZd S)rz6https://download.pytorch.org/models/vgg11-8a719046.pth crop_sizeihU ImageNet-1KgzGAQ@gx&1(V@zacc@1zacc@5V-o@g= ףp@ num_params_metrics_ops _file_sizeurlZ transformsrbN rErFrGrrr _COMMON_META IMAGENET1K_V1DEFAULTr?r?r?r@ru$ rc @ri)rz9https://download.pytorch.org/models/vgg11_bn-6002323d.pthrjrkijrmgHzQ@gp= sV@rnrogjt@rpruNrwr?r?r?r@rr{rc @ri)rz6https://download.pytorch.org/models/vgg13-19584684.pthrjrki(&rmgZd{Q@g9vOV@rnV-&@gQ@rpruNrwr?r?r?r@rr{rc @ri)rz9https://download.pytorch.org/models/vgg13_bn-abd245e5.pthrjrki(=rmg/$Q@g-V@rnr|g= ףp@rpruNrwr?r?r?r@rr{rc @seZdZedeeddiedddddid d d d Zed eeddddieddddededdid dddd Z eZ dS)rz6https://download.pytorch.org/models/vgg16-397923af.pthrjrki(+?rmgSQ@g rV@rnq= ף.@g|?5^~@rpruzIhttps://download.pytorch.org/models/vgg16_features-amdegroot-88682ab5.pth)g;pΈ?g N]?g| ?)p?r~r~)rlmeanZstdNz5https://github.com/amdegroot/ssd.pytorch#training-ssdnang#~j~@a` These weights can't be used for classification because they are missing values in the `classifier` module. Only the `features` module has valid values and can be used for feature extraction. The weights were trained using the original input standardization method as described in the paper. )rqrargrrrsrtrh) rErFrGrrr rxryrKZIMAGENET1K_FEATURESrzr?r?r?r@rsT rc @ri)rz9https://download.pytorch.org/models/vgg16_bn-6c64b313.pthrjrki(L?rmgףp= WR@g/$V@rnr}grh~@rpruNrwr?r?r?r@rr{rc @ri)rz6https://download.pytorch.org/models/vgg19-dcbb9e9d.pthrjrki(0rmgMbR@gMbV@rnoʡ3@g rh @rpruNrwr?r?r?r@r r{rc @ri)rz9https://download.pytorch.org/models/vgg19_bn-c79401a0.pthrjrki([rmgˡER@gSV@rnrg/$!@rpruNrwr?r?r?r@rr{rZ pretrained)r^T)r^r_cK t|}tdd||fi|S)apVGG-11 from `Very Deep Convolutional Networks for Large-Scale Image Recognition `__. Args: weights (:class:`~torchvision.models.VGG11_Weights`, optional): The pretrained weights to use. See :class:`~torchvision.models.VGG11_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.vgg.VGG`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.VGG11_Weights :members: rYF)rverifyrer^r_r`r?r?r@r2 rcKr)a|VGG-11-BN from `Very Deep Convolutional Networks for Large-Scale Image Recognition `__. Args: weights (:class:`~torchvision.models.VGG11_BN_Weights`, optional): The pretrained weights to use. See :class:`~torchvision.models.VGG11_BN_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.vgg.VGG`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.VGG11_BN_Weights :members: rYT)rrrerr?r?r@rLrrcKr)apVGG-13 from `Very Deep Convolutional Networks for Large-Scale Image Recognition `__. Args: weights (:class:`~torchvision.models.VGG13_Weights`, optional): The pretrained weights to use. See :class:`~torchvision.models.VGG13_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.vgg.VGG`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.VGG13_Weights :members: rZF)rrrerr?r?r@rfrrcKr)a|VGG-13-BN from `Very Deep Convolutional Networks for Large-Scale Image Recognition `__. Args: weights (:class:`~torchvision.models.VGG13_BN_Weights`, optional): The pretrained weights to use. See :class:`~torchvision.models.VGG13_BN_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.vgg.VGG`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.VGG13_BN_Weights :members: rZT)rrrerr?r?r@rrrcKr)apVGG-16 from `Very Deep Convolutional Networks for Large-Scale Image Recognition `__. Args: weights (:class:`~torchvision.models.VGG16_Weights`, optional): The pretrained weights to use. See :class:`~torchvision.models.VGG16_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.vgg.VGG`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.VGG16_Weights :members: r[F)rrrerr?r?r@r rr cKr)a|VGG-16-BN from `Very Deep Convolutional Networks for Large-Scale Image Recognition `__. Args: weights (:class:`~torchvision.models.VGG16_BN_Weights`, optional): The pretrained weights to use. See :class:`~torchvision.models.VGG16_BN_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.vgg.VGG`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.VGG16_BN_Weights :members: r[T)rrrerr?r?r@r!rr!cKr)apVGG-19 from `Very Deep Convolutional Networks for Large-Scale Image Recognition `__. Args: weights (:class:`~torchvision.models.VGG19_Weights`, optional): The pretrained weights to use. See :class:`~torchvision.models.VGG19_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.vgg.VGG`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.VGG19_Weights :members: r\F)rrrerr?r?r@r"rr"cKr)a|VGG-19_BN from `Very Deep Convolutional Networks for Large-Scale Image Recognition `__. Args: weights (:class:`~torchvision.models.VGG19_BN_Weights`, optional): The pretrained weights to use. See :class:`~torchvision.models.VGG19_BN_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.vgg.VGG`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.VGG19_BN_Weights :members: r\T)rrrerr?r?r@r#rr#)F)6 functoolsrtypingrrrrrrrBZtorch.nnr0Ztransforms._presetsr utilsr Z_apir rr_metar_utilsrr__all__rHrstrrIrJr2rTr]__annotations__rerxrrrrrrrrryrrrrr r!r"r#r?r?r?r@sn     &&"& 1*******.