U hK@sUddlmZddlmZmZmZmZmZmZddl Z ddl m Z ddl m Z ddlmZddlmZmZmZdd lmZdd lmZmZd d d ddddddddddddddgZGdd d e jZdDeeeefee jddd Z d!d"d#d"d$d$d"d%d%d"d%d%d"g d!d!d"d#d#d"d$d$d"d%d%d"d%d%d"gd!d!d"d#d#d"d$d$d$d"d%d%d%d"d%d%d%d"gd!d!d"d#d#d"d$d$d$d$d"d%d%d%d%d"d%d%d%d%d"gd&Z!eeeeeeffe"d'<eeeeeeed(d)d*Z#d+ed,d-d.Z$Gd/d d eZ%Gd0d d eZ&Gd1ddeZ'Gd2ddeZ(Gd3ddeZ)Gd4ddeZ*Gd5ddeZ+Gd6ddeZ,eed7e%j-fd8dd9d:ee%eeed;ddZ0eed7e(j-fd8dd9d:ee(eeed;d?dZ1eed7e)j-fd8dd9d:ee)eeed;d@dZ2eed7e*j-fd8dd9d:ee*eeed;dAdZ3eed7e+j-fd8dd9d:ee+eeed;dBdZ4eed7e,j-fd8dd9d:ee,eeed;dCdZ5dS)E)partial)AnycastDictListOptionalUnionN)ImageClassification)_log_api_usage_once)register_modelWeights WeightsEnum)_IMAGENET_CATEGORIES)_ovewrite_named_paramhandle_legacy_interfaceVGG 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|jdk rtj|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_weightZbiasZ constant_ BatchNorm2dZnormal_)selfr&r'r(r)m __class__H/var/www/html/venv/lib/python3.8/site-packages/torchvision/models/vgg.pyr/$s2             z VGG.__init__)xr*cCs.||}||}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#sF)cfg batch_normr*cCsg}d}|D]x}|dkr.|tjdddg7}q tt|}tj||ddd}|rl||t|tjddg7}n||tjddg7}|}q tj|S) NMr ) kernel_sizeZstrider )rQpaddingT)Zinplace)r0Z MaxPool2drrIr7r:r3r2)rMrNZlayersZ in_channelsvZconv2dr?r?r@ make_layersIs rT@rPi)ABDEcfgs)rMrNweightsprogresskwargsr*cKsl|dk r4d|d<|jddk r4t|dt|jdttt||df|}|dk rh||j|dd|S)NFr( categoriesr')rNT)r^Z check_hash)metarlenrrTr\Zload_state_dictZget_state_dict)rMrNr]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 @s>eZdZedeeddedddddid 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 rErFrGrrr _COMMON_META IMAGENET1K_V1DEFAULTr?r?r?r@rus c @s>eZdZedeeddedddddid d d d ZeZd S)rz9https://download.pytorch.org/models/vgg11_bn-6002323d.pthrhriijrkgHzQ@gp= sV@rlrmgjt@rnrsNrur?r?r?r@rs c @s>eZdZedeeddedddddid d d d ZeZd S)rz6https://download.pytorch.org/models/vgg13-19584684.pthrhrii(&rkgZd{Q@g9vOV@rlV-&@gQ@rnrsNrur?r?r?r@rs c @s>eZdZedeeddedddddid 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 c @s~eZdZedeeddedddddid d d d Zed eeddddeddddededdid 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{)rjZmeanZstdNz5https://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) rErFrGrrr rvrwrKZIMAGENET1K_FEATURESrxr?r?r?r@rsJ c @s>eZdZedeeddedddddid 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 c @s>eZdZedeeddedddddid 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 c @s>eZdZedeeddedddddid d d d ZeZd S)rz9https://download.pytorch.org/models/vgg19_bn-c79401a0.pthrhrii([rkgˡER@gSV@rlr}g/$!@rnrsNrur?r?r?r@rs Z pretrained)r]T)r]r^)r]r^r_r*cKst|}tdd||f|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 cKst|}tdd||f|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)rr~rdrr?r?r@rLs cKst|}tdd||f|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)rr~rdrr?r?r@rfs cKst|}tdd||f|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)rr~rdrr?r?r@rs cKst|}tdd||f|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)rr~rdrr?r?r@r s cKst|}tdd||f|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)rr~rdrr?r?r@r!s cKst|}tdd||f|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)rr~rdrr?r?r@r"s cKst|}tdd||f|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)rr~rdrr?r?r@r#s )F)6 functoolsrtypingrrrrrrrBZtorch.nnr0Ztransforms._presetsr utilsr Z_apir rr_metarZ_utilsrr__all__rHrstrrIrJr2rTr\__annotations__rdrvrrrrrrrrrwrrrrr r!r"r#r?r?r?r@s      &" &," 1$$$$$$$