o h K @sUddlmZddlmZmZmZmZddlZddlm Z ddl m Z ddl m Z ddlmZmZmZdd lmZdd lmZmZgd ZGd d d e jZdGdeeeefdede jfddZgdgdgdgddZ e!eeeeeffe"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)AnycastOptionalUnionN)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__\/home/www/facesmatcher.com/frenv_anti/lib/python3.10/site-packages/torchvision/models/vgg.pyr-$s:            z VGG.__init__xcCs.||}||}t|d}||}|S)Nr )r$r/torchflattenr2)r9r?r=r=r>forwardAs    z VGG.forward)r"Tr#) __name__ __module__ __qualname__r.Moduleintboolfloatr-r@ZTensorrB __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 )rOpaddingT)Zinplace)r.Z MaxPool2drrGr5r8r1r0)rKrLZlayersZ in_channelsvZconv2dr=r=r> make_layersIs  rR) @rNrNrUrNrVrNrVrVrN)rSrSrNrTrTrNrUrUrNrVrVrNrVrVrN)rSrSrNrTrTrNrUrUrUrNrVrVrVrNrVrVrVrN)rSrSrNrTrTrNrUrUrUrUrNrVrVrVrVrNrVrVrVrVrN)ABDEcfgsweightsprogresskwargscKsp|durd|d<|jddurt|dt|jdttt||dfi|}|dur6||j|dd|S)NFr& categoriesr%)rLT)r]Z check_hash)metarlenrrRr[Zload_state_dictZget_state_dict)rKrLr\r]r^modelr=r=r>_vggbsrc) rdzUhttps://github.com/pytorch/vision/tree/main/references/classification#alexnet-and-vggzNThese weights were trained from scratch by using a simplified training recipe.)Zmin_sizer_recipe_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 transformsr`N rCrDrEr rr _COMMON_META IMAGENET1K_V1DEFAULTr=r=r=r>ru$ rc @rg)rz9https://download.pytorch.org/models/vgg11_bn-6002323d.pthrhriijrkgHzQ@gp= sV@rlrmgjt@rnrsNrur=r=r=r>rryrc @rg)rz6https://download.pytorch.org/models/vgg13-19584684.pthrhrii(&rkgZd{Q@g9vOV@rlV-&@gQ@rnrsNrur=r=r=r>rryrc @rg)rz9https://download.pytorch.org/models/vgg13_bn-abd245e5.pthrhrii(=rkg/$Q@g-V@rlrzg= ףp@rnrsNrur=r=r=r>rryrc @seZdZedeeddiedddddid d d d Zed eeddddieddddededdid dddd Z eZ dS)rz6https://download.pytorch.org/models/vgg16-397923af.pthrhrii(+?rkgSQ@g rV@rlq= ף.@g|?5^~@rnrszIhttps://download.pytorch.org/models/vgg16_features-amdegroot-88682ab5.pth)g;pΈ?g N]?g| ?)p?r|r|)rjmeanZstdNz5https://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. )ror_rerprqrrrf) rCrDrEr rrrvrwrIZIMAGENET1K_FEATURESrxr=r=r=r>rsT rc @rg)rz9https://download.pytorch.org/models/vgg16_bn-6c64b313.pthrhrii(L?rkgףp= WR@g/$V@rlr{grh~@rnrsNrur=r=r=r>rryrc @rg)rz6https://download.pytorch.org/models/vgg19-dcbb9e9d.pthrhrii(0rkgMbR@gMbV@rloʡ3@g rh @rnrsNrur=r=r=r>r ryrc @rg)rz9https://download.pytorch.org/models/vgg19_bn-c79401a0.pthrhrii([rkgˡER@gSV@rlrg/$!@rnrsNrur=r=r=r>rryrZ 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: rWF)rverifyrcr\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: rWT)rrrcrr=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: rXF)rrrcrr=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: rXT)rrrcrr=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: rYF)rrrcrr=r=r>rrrcKr)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: rYT)rrrcrr=r=r>rrrcKr)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: rZF)rrrcrr=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: rZT)rrrcrr=r=r>r!rr!)F)6 functoolsrtypingrrrrr@Ztorch.nnr.Ztransforms._presetsrutilsr Z_apir r r _metarZ_utilsrr__all__rFrliststrrGrHr0rRr[dict__annotations__rcrvrrrrrrrrrwrrrrrrr r!r=r=r=r>sn    &&"& 1*******.