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)pfan_outrelu)mode nonlinearityrr g{Gz?)super__init__r r$nnAdaptiveAvgPool2davgpool SequentialLinearReLUDropout classifiermodules isinstanceConv2dinitkaiming_normal_weightbias constant_ BatchNorm2dnormal_)selfr$r%r&r'm __class__Q/var/www/html/scripts/venv/lib/python3.10/site-packages/torchvision/models/vgg.pyr0$s:            z VGG.__init__xcCs.||}||}t|d}||}|S)Nr )r$r3torchflattenr8)rCrIrGrGrHforwardAs    z VGG.forward)r"Tr#) __name__ __module__ __qualname__r1Moduleintboolfloatr0rJTensorrL __classcell__rGrGrErHr#srFcfg batch_normr(cCsg}d}|D]<}|dkr|tjdddg7}qtt|}tj||ddd}|r6||t|tjddg7}n ||tjddg7}|}qtj|S) NMr) kernel_sizestrider )rZpaddingT)inplace)r1 MaxPool2drrQr;rAr6r4)rVrWlayers in_channelsvconv2drGrGrH make_layersIs  rc) @rYrYrfrYrgrYrgrgrY)rdrdrYrererYrfrfrYrgrgrYrgrgrY)rdrdrYrererYrfrfrfrYrgrgrgrYrgrgrgrY)rdrdrYrererYrfrfrfrfrYrgrgrgrgrYrgrgrgrgrY)ABDEcfgsweightsprogresskwargscKsp|durd|d<|jddurt|dt|jdttt||dfi|}|dur6||j|dd|S)NFr& categoriesr%)rWT)rn check_hash)metarlenrrcrlload_state_dictget_state_dict)rVrWrmrnromodelrGrGrH_vggbsrw) rxzUhttps://github.com/pytorch/vision/tree/main/references/classification#alexnet-and-vggzNThese weights were trained from scratch by using a simplified training recipe.)min_sizerprecipe_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_sizeurl transformsrrN rMrNrOr rr _COMMON_META IMAGENET1K_V1DEFAULTrGrGrGrHru$ rc @r|)rz9https://download.pytorch.org/models/vgg11_bn-6002323d.pthr}r~ijrgHzQ@gp= sV@rrgjt@rrNrrGrGrGrHrrrc @r|)rz6https://download.pytorch.org/models/vgg13-19584684.pthr}r~i(&rgZd{Q@g9vOV@rV-&@gQ@rrNrrGrGrGrHrrrc @r|)rz9https://download.pytorch.org/models/vgg13_bn-abd245e5.pthr}r~i(=rg/$Q@g-V@rrg= ףp@rrNrrGrGrGrHrrrc @seZdZedeeddiedddddid d d d Zed eeddddieddddededdid dddd Z eZ dS)rz6https://download.pytorch.org/models/vgg16-397923af.pthr}r~i(+?rgSQ@g rV@rq= ף.@g|?5^~@rrzIhttps://download.pytorch.org/models/vgg16_features-amdegroot-88682ab5.pth)g;pΈ?g N]?g| ?)p?rr)rmeanstdNz5https://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. )rrprzrrrr{) rMrNrOr rrrrrSIMAGENET1K_FEATURESrrGrGrGrHrsT rc @r|)rz9https://download.pytorch.org/models/vgg16_bn-6c64b313.pthr}r~i(L?rgףp= WR@g/$V@rrgrh~@rrNrrGrGrGrHrrrc @r|)rz6https://download.pytorch.org/models/vgg19-dcbb9e9d.pthr}r~i(0rgMbR@gMbV@roʡ3@g rh @rrNrrGrGrGrHr rrc @r|)rz9https://download.pytorch.org/models/vgg19_bn-c79401a0.pthr}r~i([rgˡER@gSV@rrg/$!@rrNrrGrGrGrHrrr pretrained)rmT)rmrncK 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: rhF)rverifyrwrmrnrorGrGrHr2 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: rhT)rrrwrrGrGrHrLrrcKr)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: riF)rrrwrrGrGrHrfrrcKr)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: riT)rrrwrrGrGrHrrrcKr)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: rjF)rrrwrrGrGrHrrrcKr)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: rjT)rrrwrrGrGrHrrrcKr)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: rkF)rrrwrrGrGrHr 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: rkT)rrrwrrGrGrHr!rr!)F)6 functoolsrtypingrrrrrJtorch.nnr1transforms._presetsrutilsr _apir r r _metar_utilsrr__all__rPrliststrrQrRr4rcrldict__annotations__rwrrrrrrrrrrrrrrrrr r!rGrGrGrHsn    &&"& 1*******.