o V hK @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)pfan_outrelu)mode nonlinearityrr g{Gz?)super__init__r r&nnAdaptiveAvgPool2davgpool SequentialLinearReLUDropout classifiermodules isinstanceConv2dinitkaiming_normal_weightbias constant_ BatchNorm2dnormal_)selfr&r'r(r)m __class__J/var/www/vscode/kcb/lib/python3.10/site-packages/torchvision/models/vgg.pyr2$s:            z VGG.__init__xcCs.||}||}t|d}||}|S)Nr )r&r5torchflattenr:)rErKrIrIrJforwardAs    z VGG.forward)r$Tr%) __name__ __module__ __qualname__r3Moduleintboolfloatr2rLTensorrN __classcell__rIrIrGrJr#srFcfg batch_normr*cCsg}d}|D]<}|dkr|tjdddg7}qtt|}tj||ddd}|r6||t|tjddg7}n ||tjddg7}|}qtj|S) NMr ) kernel_sizestrider )r\paddingT)inplace)r3 MaxPool2drrSr=rCr8r6)rXrYlayers in_channelsvconv2drIrIrJ make_layersIs  re) @r[r[rhr[rir[ririr[)rfrfr[rgrgr[rhrhr[ririr[ririr[)rfrfr[rgrgr[rhrhrhr[riririr[riririr[)rfrfr[rgrgr[rhrhrhrhr[ririririr[ririririr[)ABDEcfgsweightsprogresskwargscKsp|durd|d<|jddurt|dt|jdttt||dfi|}|dur6||j|dd|S)NFr( categoriesr')rYT)rp check_hash)metarlenrrernload_state_dictget_state_dict)rXrYrorprqmodelrIrIrJ_vggbsry) rzzUhttps://github.com/pytorch/vision/tree/main/references/classification#alexnet-and-vggzNThese weights were trained from scratch by using a simplified training recipe.)min_sizerrrecipe_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 transformsrtN rOrPrQrrr _COMMON_META IMAGENET1K_V1DEFAULTrIrIrIrJru$ rc @r~)rz9https://download.pytorch.org/models/vgg11_bn-6002323d.pthrrijrgHzQ@gp= sV@rrgjt@rrNrrIrIrIrJrrrc @r~)rz6https://download.pytorch.org/models/vgg13-19584684.pthrri(&rgZd{Q@g9vOV@rV-&@gQ@rrNrrIrIrIrJrrrc @r~)rz9https://download.pytorch.org/models/vgg13_bn-abd245e5.pthrri(=rg/$Q@g-V@rrg= ףp@rrNrrIrIrIrJrrrc @seZdZedeeddiedddddid d d d Zed eeddddieddddededdid dddd Z eZ dS)rz6https://download.pytorch.org/models/vgg16-397923af.pthrri(+?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. )rrrr|rrrr}) rOrPrQrrr rrrUIMAGENET1K_FEATURESrrIrIrIrJrsT rc @r~)rz9https://download.pytorch.org/models/vgg16_bn-6c64b313.pthrri(L?rgףp= WR@g/$V@rrgrh~@rrNrrIrIrIrJrrrc @r~)rz6https://download.pytorch.org/models/vgg19-dcbb9e9d.pthrri(0rgMbR@gMbV@roʡ3@g rh @rrNrrIrIrIrJr rrc @r~)rz9https://download.pytorch.org/models/vgg19_bn-c79401a0.pthrri([rgˡER@gSV@rrg/$!@rrNrrIrIrIrJrrr pretrained)roT)rorpcK 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: rjF)rverifyryrorprqrIrIrJr2 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: rjT)rrryrrIrIrJrLrrcKr)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: rkF)rrryrrIrIrJrfrrcKr)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: rkT)rrryrrIrIrJrrrcKr)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: rlF)rrryrrIrIrJr 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: rlT)rrryrrIrIrJr!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: rmF)rrryrrIrIrJr"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: rmT)rrryrrIrIrJr#rr#)F)6 functoolsrtypingrrrrrrrLtorch.nnr3transforms._presetsr utilsr _apir rr_metar_utilsrr__all__rRrstrrSrTr6rern__annotations__ryrrrrrrrrrrrrrrr r!r"r#rIrIrIrJsn     &&"& 1*******.