a dK@s^UddlmZddlmZmZmZmZmZmZddl Z ddl m Z ddl m Z ddlmZddlmZmZmZdd lmZdd lmZmZgd ZGd d d e jZdDeeeefee jdddZ gdgdgdgddZ!eeeeeeffe"d<eeeeeeedddZ#dedddZ$Gdd d eZ%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-d.d.eZ,eed/e%j-fd0dd1d2ee%eeed3d4d5Z.eed/e&j-fd0dd1d2ee&eeed3d6d7Z/eed/e'j-fd0dd1d2ee'eeed3d8d9Z0eed/e(j-fd0dd1d2ee(eeed3d:d;Z1eed/e)j-fd0dd1d2ee)eeed3dd?Z3eed/e+j-fd0dd1d2ee+eeed3d@dAZ4eed/e,j-fd0dd1d2ee,eeed3dBdCZ5dS)E)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_bncsBeZdZd ejeeeddfdd Ze j e j dd d Z Z S) rT?N)features num_classes init_weightsdropoutreturnc s tt|||_td|_ttddt dtj |dtddt dtj |dtd||_ |r| D]}t |tjrtjj|jddd|jdurtj|jd q~t |tjrtj|jd tj|jd q~t |tjr~tj|jd d tj|jd q~dS) N)r+ibiT)pZfan_outZrelu)modeZ nonlinearityrr g{Gz?)super__init__r r&nnZAdaptiveAvgPool2davgpool SequentialZLinearReLUZDropout classifiermodules isinstanceConv2dinitZkaiming_normal_ZweightZbiasZ constant_ BatchNorm2dZnormal_)selfr&r'r(r)m __class___/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torchvision/models/vgg.pyr/$s2             z VGG.__init__)xr*cCs.||}||}t|d}||}|S)Nr )r&r1torchflattenr4)r:r@r>r>r?forwardAs     z VGG.forward)r$Tr%) __name__ __module__ __qualname__r0Moduleintboolfloatr/rAZTensorrC __classcell__r>r>r<r?r#s  rF)cfg batch_normr*cCsg}d}|D]x}|dkr.|tjdddg7}q tt|}tj||ddd}|rl||t|tjddg7}n||tjddg7}|}q tj|S) NMr ) kernel_sizeZstrider )rPpaddingT)Zinplace)r0Z MaxPool2drrHr7r9r3r2)rLrMZlayersZ in_channelsvZconv2dr>r>r? make_layersIs rS) @rOrOrVrOrWrOrWrWrO)rTrTrOrUrUrOrVrVrOrWrWrOrWrWrO)rTrTrOrUrUrOrVrVrVrOrWrWrWrOrWrWrWrO)rTrTrOrUrUrOrVrVrVrVrOrWrWrWrWrOrWrWrWrWrO)ABDEcfgs)rLrMweightsprogresskwargsr*cKsn|dur4d|d<|jddur4t|dt|jdttt||dfi|}|durj||j|d|S)NFr( categoriesr')rM)r^)metarlenrrSr\Zload_state_dictZget_state_dict)rLrMr]r^r_modelr>r>r?_vggbsrd) rezUhttps://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 @sBeZdZedeeddiedddddid 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 transformsraN rDrErFrrr _COMMON_META IMAGENET1K_V1DEFAULTr>r>r>r?rus" rc @sBeZdZedeeddiedddddid d d d ZeZd S)rz9https://download.pytorch.org/models/vgg11_bn-6002323d.pthrhriijrkgHzQ@gp= sV@rlrmgjt@rnrsNrur>r>r>r?rs" rc @sBeZdZedeeddiedddddid d d d ZeZd S)rz6https://download.pytorch.org/models/vgg13-19584684.pthrhrii(&rkgZd{Q@g9vOV@rlV-&@gQ@rnrsNrur>r>r>r?rs" rc @sBeZdZedeeddiedddddid d d d ZeZd S)rz9https://download.pytorch.org/models/vgg13_bn-abd245e5.pthrhrii(=rkg/$Q@g-V@rlryg= ףp@rnrsNrur>r>r>r?rs" rc @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`rfrprqrrrg) rDrErFrrr rvrwrJZIMAGENET1K_FEATURESrxr>r>r>r?rsR rc @sBeZdZedeeddiedddddid d d d ZeZd S)rz9https://download.pytorch.org/models/vgg16_bn-6c64b313.pthrhrii(L?rkgףp= WR@g/$V@rlrzgrh~@rnrsNrur>r>r>r?rs" rc @sBeZdZedeeddiedddddid d d d ZeZd S)rz6https://download.pytorch.org/models/vgg19-dcbb9e9d.pthrhrii(0rkgMbR@gMbV@rloʡ3@g rh @rnrsNrur>r>r>r?r s" rc @sBeZdZedeeddiedddddid d d d ZeZd S)rz9https://download.pytorch.org/models/vgg19_bn-c79401a0.pthrhrii([rkgˡER@gSV@rlr~g/$!@rnrsNrur>r>r>r?rs" rZ pretrained)r]T)r]r^)r]r^r_r*cKs 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: rXF)rverifyrdr]r^r_r>r>r?r2s rcKs t|}tdd||fi|S)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: rXT)rrrdrr>r>r?rLs rcKs t|}tdd||fi|S)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: rYF)rrrdrr>r>r?rfs rcKs t|}tdd||fi|S)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: rYT)rrrdrr>r>r?rs rcKs t|}tdd||fi|S)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: rZF)rrrdrr>r>r?r s r cKs t|}tdd||fi|S)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: rZT)rrrdrr>r>r?r!s r!cKs t|}tdd||fi|S)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)rrrdrr>r>r?r"s r"cKs t|}tdd||fi|S)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)rrrdrr>r>r?r#s r#)F)6 functoolsrtypingrrrrrrrAZtorch.nnr0Ztransforms._presetsr utilsr Z_apir rrZ_metar_utilsrr__all__rGrstrrHrIr2rSr\__annotations__rdrvrrrrrrrrrwrrrrr r!r"r#r>r>r>r?sl     &"" 1$$$$$$$