U (dJ@sUddlmZddlmZmZmZmZmZmZddl Z ddl m Z ddl m Z ddlmZddlmZmZdd lmZdd lmZmZd d d ddddddddddddddgZGdd d e jZdFeeeefee jddd Zd!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+ed7e$j,fd8dd9d:ee$eeed;ddZ/ed7e'j,fd8dd9d:ee'eeed;d?dZ0ed7e(j,fd8dd9d:ee(eeed;d@dZ1ed7e)j,fd8dd9d:ee)eeed;dAdZ2ed7e*j,fd8dd9d:ee*eeed;dBdZ3ed7e+j,fd8dd9d:ee+eeed;dCdZ4ddDlm5Z5e5e$j,j6e&j,j6e(j,j6e*j,j6e%j,j6e'j,j6e)j,j6e+j,j6dEZ7dS)G)partial)UnionListDictAnyOptionalcastN)ImageClassification)_log_api_usage_once) WeightsEnumWeights)_IMAGENET_CATEGORIES)handle_legacy_interface_ovewrite_named_paramVGG 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_ZweightZbiasZ constant_ BatchNorm2dZnormal_)selfr%r&r'r(m __class__:/tmp/pip-unpacked-wheel-vx7f76es/torchvision/models/vgg.pyr.$s2             z VGG.__init__)xr)cCs.||}||}t|d}||}|S)Nr )r%r0torchflattenr3)r9r?r=r=r>forwardAs     z VGG.forward)r#Tr$) __name__ __module__ __qualname__r/Moduleintboolfloatr.r@ZTensorrB __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 )rOpaddingT)Zinplace)r/Z MaxPool2drrGr6r8r2r1)rKrLZlayersZ in_channelsvZconv2dr=r=r> make_layersIs rR@rNi)ABDEcfgs)rKrLweightsprogresskwargsr)cKsj|dk r4d|d<|jddk r4t|dt|jdttt||df|}|dk rf||j|d|S)NFr' categoriesr&)rL)r\)metarlenrrRrZZload_state_dictZget_state_dict)rKrLr[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 @s:eZdZedeeddedddddid d ZeZd S) rz6https://download.pytorch.org/models/vgg11-8a719046.pth crop_sizeihU ImageNet-1KgzGAQ@gx&1(V@zacc@1zacc@5 num_params_metricsurlZ transformsr_N rCrDrErrr _COMMON_META IMAGENET1K_V1DEFAULTr=r=r=r>rus c @s:eZdZedeeddedddddid d ZeZd S) rz9https://download.pytorch.org/models/vgg11_bn-6002323d.pthrfrgijrigHzQ@gp= sV@rjrkrnNrpr=r=r=r>rs c @s:eZdZedeeddedddddid d ZeZd S) rz6https://download.pytorch.org/models/vgg13-19584684.pthrfrgi(&rigZd{Q@g9vOV@rjrkrnNrpr=r=r=r>rs c @s:eZdZedeeddedddddid d ZeZd S) rz9https://download.pytorch.org/models/vgg13_bn-abd245e5.pthrfrgi(=rig/$Q@g-V@rjrkrnNrpr=r=r=r>rs c @sveZdZedeeddedddddid d Zed eedd d deddddededdiddd Z eZ dS)rz6https://download.pytorch.org/models/vgg16-397923af.pthrfrgi(+?rigSQ@g rV@rjrkrnzIhttps://download.pytorch.org/models/vgg16_features-amdegroot-88682ab5.pth)g;pΈ?g N]?g| ?)p?rtrt)rhZmeanZstdNz5https://github.com/amdegroot/ssd.pytorch#training-ssdnana` 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. )rlr^rdrmre) rCrDrErrr rqrrrIZIMAGENET1K_FEATURESrsr=r=r=r>rsB c @s:eZdZedeeddedddddid d ZeZd S) rz9https://download.pytorch.org/models/vgg16_bn-6c64b313.pthrfrgi(L?rigףp= WR@g/$V@rjrkrnNrpr=r=r=r>rs c @s:eZdZedeeddedddddid d ZeZd S) rz6https://download.pytorch.org/models/vgg19-dcbb9e9d.pthrfrgi(0rigMbR@gMbV@rjrkrnNrpr=r=r=r>rs c @s:eZdZedeeddedddddid d ZeZd S) rz9https://download.pytorch.org/models/vgg19_bn-c79401a0.pthrfrgi([rigˡER@gSV@rjrkrnNrpr=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: rVF)rverifyrbr[r\r]r=r=r>r s 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: rVT)rrvrbrwr=r=r>r9s 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: rWF)rrvrbrwr=r=r>rRs 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: rWT)rrvrbrwr=r=r>rks 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: rXF)rrvrbrwr=r=r>rs 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: rXT)rrvrbrwr=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: rYF)rrvrbrwr=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: rYT)rrvrbrwr=r=r>r"s ) _ModelURLs)rrrr!rrr r")F)8 functoolsrtypingrrrrrrr@Ztorch.nnr/Ztransforms._presetsr utilsr Z_apir rZ_metar_utilsrr__all__rFrstrrGrHr1rRrZ__annotations__rbrqrrrrrrrrrrrrrrrr r!r"rxroZ model_urlsr=r=r=r>s      &" &," -""""""""