jM XUddlmZddlmZmZmZmZmZmZddl Z ddl m Z ddl m Z ddlmZddlmZmZmZdd lmZdd lmZmZgd ZGd d e jZd=deeeefdede jfdZ gdgdgdgddZ!eeeeeeffe"d<dededeedededef dZ#deddd Z$Gd!d"eZ%Gd#d$eZ&Gd%d&eZ'Gd'd(eZ(Gd)d*eZ)Gd+d,eZ*Gd-d.eZ+Gd/d0eZ,eed1e%j-f2dd3d4dee%dededefd5Z.eed1e&j-f2dd3d4dee&dededefd6Z/eed1e'j-f2dd3d4dee'dededefd7Z0eed1e(j-f2dd3d4dee(dededefd8Z1eed1e)j-f2dd3d4dee)dededefd9Z2eed1e*j-f2dd3d4dee*dededefd:Z3eed1e+j-f2dd3d4dee+dededefd;Z4eed1e,j-f2dd3d4dee,dededefd<Z5dS)>)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 feZdZ d dejdedededd f fd Zd e j de j fd Z xZ S)rT?features num_classes init_weightsdropoutreturnNc tt|||_t jd|_t jt jddt j dt j |t jddt j dt j |t jd||_ |rQ| D]=}t|tjrTtj|jdd|j%tj|jd qt|tjrKtj|jd tj|jd t|tjrKtj|jd d tj|jd =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__s GC:\Projects\Historical\venv\Lib\site-packages\torchvision/models/vgg.pyr5z VGG.__init__$s D!!!  +F33 - Ik4 ( ( GDMM J ! ! ! IdD ! ! GDMM J ! ! ! IdK ( (    1\\^^ 1 1a++ 1G++AH9SY+ZZZv)))!&!4442>221G%%ah222G%%afa000029--1GOOAHa666G%%afa000 1 1 1 1xc||}||}tj|d}||}|S)Nr )r(r8torchflattenr=)rHrMs rKforwardz VGG.forwardAsI MM!   LLOO M!Q   OOA  rL)r&Tr') __name__ __module__ __qualname__r6Moduleintboolfloatr5rOTensorrQ __classcell__)rJs@rKrr#shk11 1031JN1`e1 111111:%,rLrFcfg batch_normr,cdg}d}|D]}|dkr|tjddgz }#tt|}tj||dd}|r.||tj|tjdgz }n||tjdgz }|}tj|S) NMr ) kernel_sizestrider )r`paddingT)inplace)r6 MaxPool2drrVr@rFr;r9)r[r\layers in_channelsvconv2ds rK make_layersriIs FK    88 r|!<<<= =FFS! AY{A1aHHHF :62>!#4#4bgd6K6K6KLL6274#8#8#899KK =& !!rL) @r_r_rlr_rmr_rmrmr_)rjrjr_rkrkr_rlrlr_rmrmr_rmrmr_)rjrjr_rkrkr_rlrlrlr_rmrmrmr_rmrmrmr_)rjrjr_rkrkr_rlrlrlrlr_rmrmrmrmr_rmrmrmrmr_)ABDEcfgsweightsprogresskwargsc &|;d|d<|jd)t|dt|jdtt t ||fi|}|*|||d|S)NFr* categoriesr))r\T)rt check_hash)metarlenrrirrload_state_dictget_state_dict)r[r\rsrtrumodels rK_vggr~bs!&~ < % 1 !&-W\,=W9X9X Y Y Y  DI*=== H H H HE g44hSW4XXYYY LrL) rzUhttps://github.com/pytorch/vision/tree/main/references/classification#alexnet-and-vggzNThese weights were trained from scratch by using a simplified training recipe.)min_sizerwrecipe_docsc `eZdZedeediedddddid 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 transformsryN rRrSrTrrr _COMMON_META IMAGENET1K_V1DEFAULTrLrKrrusG D7.#>>>    ###        M GGGrLrc `eZdZedeediedddddid d d  ZeZd S)rz9https://download.pytorch.org/models/vgg11_bn-6002323d.pthrrijrgHzQ@gp= sV@rrgjt@rrNrrrLrKrrG G7.#>>>    ###   !    M GGGrLrc `eZdZedeediedddddid d d  ZeZd S)rz6https://download.pytorch.org/models/vgg13-19584684.pthrri(&rgZd{Q@g9vOV@rV-&@gQ@rrNrrrLrKrrG D7.#>>>    ###   !    M GGGrLrc `eZdZedeediedddddid d d  ZeZd S)rz9https://download.pytorch.org/models/vgg13_bn-abd245e5.pthrri(=rg/$Q@g-V@rrg= ףp@rrNrrrLrKrrsG G7.#>>>    ###        M GGGrLrc eZdZedeediedddddid d d  Zed eedddieddddededdid ddd 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. )rrwrrrrr) rRrSrTrrr rrrXIMAGENET1K_FEATURESrrrLrKrrsG D7.#>>>    ###   !    M "' W7 ,7      #M"U5\\"U5\\   !   :GGGrLrc `eZdZedeediedddddid d d  ZeZd S)rz9https://download.pytorch.org/models/vgg16_bn-6c64b313.pthrri(L?rgףp= WR@g/$V@rrgrh~@rrNrrrLrKrrrrLrc `eZdZedeediedddddid d d  ZeZd S)rz6https://download.pytorch.org/models/vgg19-dcbb9e9d.pthrri(0rgMbR@gMbV@roʡ3@g rh @rrNrrrLrKrr rrLrc `eZdZedeediedddddid d d  ZeZd S)rz9https://download.pytorch.org/models/vgg19_bn-c79401a0.pthrri([rgˡER@gSV@rrg/$!@rrNrrrLrKrrsG G7.#>>>    ###   !    M GGGrLr pretrained)rsT)rsrtc Vt|}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: rnF)rverifyr~rsrtrus rKrr23*""7++G UGX 8 8 8 88rLc Vt|}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: rnT)rrr~rs rKrrL3*%%g..G T7H 7 7 7 77rLc Vt|}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: roF)rrr~rs rKrrfrrLc Vt|}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: roT)rrr~rs rKr r rrLc Vt|}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: rpF)rrr~rs rKr!r!rrLc Vt|}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: rpT)rrr~rs rKr"r"rrLc Vt|}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: rqF)rrr~rs rKr#r#rrLc Vt|}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: rqT)rrr~rs rKr$r$rrL)F)6 functoolsrtypingrrrrrr rOtorch.nnr6transforms._presetsr utilsr _apirrr_metar_utilsrr__all__rUrstrrVrWr9rirr__annotations__r~rrrrrrrrrrrrrr r!r"r#r$rrLrKrsQ9999999999999999 555555''''''6666666666''''''BBBBBBBB   *#####")###L""T%S/*""""""$ J I I R R R a a a p p p **d3U38_% %&cth{.Ct_bgj&e a  K({(K({(.....K...b{(K({(, 0K!LMMM04t999h}-99WZ9_b999NM90,0@0N!OPPP6:T888"238d8]`8eh888QP80, 0K!LMMM04t999h}-99WZ9_b999NM90,0@0N!OPPP6:T888"238d8]`8eh888QP80, 0K!LMMM04t999h}-99WZ9_b999NM90,0@0N!OPPP6:T888"238d8]`8eh888QP80, 0K!LMMM04t999h}-99WZ9_b999NM90,0@0N!OPPP6:T888"238d8]`8eh888QP888rL