a h K@sVUddlmZddlmZmZmZmZddlZddlm Z ddl m Z ddl m Z ddlmZmZmZdd lmZdd lmZmZgd ZGd d d e jZdDeeeefee jdddZgdgdgdgddZ e!eeeeeffe"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)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_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__D/var/www/auris/lib/python3.9/site-packages/torchvision/models/vgg.pyr-$s2             z VGG.__init__)xr(cCs.||}||}t|d}||}|S)Nr )r$r/torchflattenr2)r8r>r<r<r=forwardAs     z VGG.forward)r"Tr#) __name__ __module__ __qualname__r.Moduleintboolfloatr-r?ZTensorrA __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 )rNpaddingT)Zinplace)r.Z MaxPool2drrFr5r7r1r0)rJrKZlayersZ in_channelsvZconv2dr<r<r= make_layersIs rQ) @rMrMrTrMrUrMrUrUrM)rRrRrMrSrSrMrTrTrMrUrUrMrUrUrM)rRrRrMrSrSrMrTrTrTrMrUrUrUrMrUrUrUrM)rRrRrMrSrSrMrTrTrTrTrMrUrUrUrUrMrUrUrUrUrM)ABDEcfgs)rJrKweightsprogresskwargsr(cKsp|dur4d|d<|jddur4t|dt|jdttt||dfi|}|durl||j|dd|S)NFr& categoriesr%)rKT)r\Z check_hash)metarlenrrQrZZload_state_dictZget_state_dict)rJrKr[r\r]modelr<r<r=_vggbsrb) rczUhttps://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 transformsr_N rBrCrDr rr _COMMON_META IMAGENET1K_V1DEFAULTr<r<r<r=rus" rc @sBeZdZedeeddiedddddid d d d ZeZd S)rz9https://download.pytorch.org/models/vgg11_bn-6002323d.pthrfrgijrigHzQ@gp= sV@rjrkgjt@rlrqNrsr<r<r<r=rs" rc @sBeZdZedeeddiedddddid d d d ZeZd S)rz6https://download.pytorch.org/models/vgg13-19584684.pthrfrgi(&rigZd{Q@g9vOV@rjV-&@gQ@rlrqNrsr<r<r<r=rs" rc @sBeZdZedeeddiedddddid d d d ZeZd S)rz9https://download.pytorch.org/models/vgg13_bn-abd245e5.pthrfrgi(=rig/$Q@g-V@rjrwg= ףp@rlrqNrsr<r<r<r=rs" rc @seZdZedeeddiedddddid d d d Zed eeddddieddddededdid dddd Z eZ dS)rz6https://download.pytorch.org/models/vgg16-397923af.pthrfrgi(+?rigSQ@g rV@rjq= ף.@g|?5^~@rlrqzIhttps://download.pytorch.org/models/vgg16_features-amdegroot-88682ab5.pth)g;pΈ?g N]?g| ?)p?ryry)rhmeanZstdNz5https://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. )rmr^rdrnrorpre) rBrCrDr rrrtrurHZIMAGENET1K_FEATURESrvr<r<r<r=rsR rc @sBeZdZedeeddiedddddid d d d ZeZd S)rz9https://download.pytorch.org/models/vgg16_bn-6c64b313.pthrfrgi(L?rigףp= WR@g/$V@rjrxgrh~@rlrqNrsr<r<r<r=rs" rc @sBeZdZedeeddiedddddid d d d ZeZd S)rz6https://download.pytorch.org/models/vgg19-dcbb9e9d.pthrfrgi(0rigMbR@gMbV@rjoʡ3@g rh @rlrqNrsr<r<r<r=r s" rc @sBeZdZedeeddiedddddid d d d ZeZd S)rz9https://download.pytorch.org/models/vgg19_bn-c79401a0.pthrfrgi([rigˡER@gSV@rjr|g/$!@rlrqNrsr<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: rVF)rverifyrbr[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: rVT)rr}rbr~r<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: rWF)rr}rbr~r<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: rWT)rr}rbr~r<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: rXF)rr}rbr~r<r<r=rs rcKs 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: rXT)rr}rbr~r<r<r=rs rcKs 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: rYF)rr}rbr~r<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: rYT)rr}rbr~r<r<r=r!s r!)F)6 functoolsrtypingrrrrr?Ztorch.nnr.Ztransforms._presetsrutilsr Z_apir r r Z_metar_utilsrr__all__rErliststrrFrGr0rQrZdict__annotations__rbrtrrrrrrrrrurrrrrrr r!r<r<r<r=sl    &"" 1$$$$$$$