khK pUddlmZddlmZmZmZmZmZmZddl Z ddl m Z ddl m Z ddlmZddlmZmZmZdd lmZdd lmZmZgd ZGd d e j4ZdfdZ gdgdgdgddZ!eeeeeeffe"d<dededeedededef dZ#dedddZ$Gd d!eZ%Gd"d#eZ&Gd$d%eZ'Gd&d'eZ(Gd(d)eZ)Gd*d+eZ*Gd,d-eZ+Gd.d/eZ,eed0e%jZf1dd2d3dee%dededefd4Z.eed0e&jZf1dd2d3dee&dededefd5Z/eed0e'jZf1dd2d3dee'dededefd6Z0eed0e(jZf1dd2d3dee(dededefd7Z1eed0e)jZf1dd2d3dee)dededefd8Z2eed0e*jZf1dd2d3dee*dededefd9Z3eed0e+jZf1dd2d3dee+dededefd:Z4eed0e,jZf1dd2d3dee,dededefd;Z5y)=)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 eZdZ d dejdedededdf fd Zde jde jfd Z xZ S) rfeatures num_classes init_weightsdropoutreturnNc t|t|||_t j d|_t jt jddt jdt j|t jddt jdt j|t jd||_ |rv|jD]a}t|tjrdtjj!|j"dd|j$Wtjj'|j$d t|tj(rUtjj'|j"d tjj'|j$d t|tjs tjj+|j"d d tjj'|j$d dyy) 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 R/var/www/teggl/fontify/venv/lib/python3.12/site-packages/torchvision/models/vgg.pyr3z VGG.__init__$s D!  ++F3 -- IIk4 ( GGDM JJ ! IIdD ! GGDM JJ ! IIdK (  \\^ 1a+GG++AHH9SY+Zvv)))!&&!42>>2GG%%ahh2GG%%affa0299-GGOOAHHa6GG%%affa0 1 xc|j|}|j|}tj|d}|j |}|S)Nr )r&r6torchflattenr;)rFrKs rIforwardz VGG.forwardAs@ MM!  LLO MM!Q  OOA rJ)iTg?) __name__ __module__ __qualname__r4Moduleintboolfloatr3rMTensorrO __classcell__)rHs@rIrr#sRhk1 1031JN1`e1 1:%,,rJrcfg batch_normr*cng}d}|D]}|dk(r|tjddgz }$tt|}tj||dd}|r0||tj |tj dgz }n||tj dgz }|}tj|S) NMr ) kernel_sizestrider )r^paddingT)inplace)r4 MaxPool2drrTr>rDr9r7)rYrZlayers in_channelsvconv2ds rI make_layersrgIs FK   8 r||!<= =FS! AYY{A1aHF62>>!#4bggd6KLL62774#899K  ==& !!rJ) @r]r]rjr]rkr]rkrkr])rhrhr]ririr]rjrjr]rkrkr]rkrkr])rhrhr]ririr]rjrjrjr]rkrkrkr]rkrkrkr])rhrhr]ririr]rjrjrjrjr]rkrkrkrkr]rkrkrkrkr])ABDEcfgsweightsprogresskwargsc |7d|d<|jd#t|dt|jdtt t ||fi|}|"|j |j|d|S)NFr( categoriesr')rZT)rr check_hash)metarlenrrgrpload_state_dictget_state_dict)rYrZrqrrrsmodels rI_vggr|bs!&~ << % 1 !&-W\\,=W9X Y  DI*= H HE g44hSW4XY LrJ) r}zUhttps://github.com/pytorch/vision/tree/main/references/classification#alexnet-and-vggzNThese weights were trained from scratch by using a simplified training recipe.)min_sizerurecipe_docsc ReZdZedeediedddddid d d  ZeZy )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 transformsrwN rPrQrRrrr _COMMON_META IMAGENET1K_V1DEFAULTrJrIrrusQ D.#>    ###     M GrJrc ReZdZedeediedddddid d d  ZeZy )rz9https://download.pytorch.org/models/vgg11_bn-6002323d.pthrrijrgHzQ@gp= sV@rrgjt@rrNrrrJrIrrQ G.#>    ###  !  M GrJrc ReZdZedeediedddddid d d  ZeZy )rz6https://download.pytorch.org/models/vgg13-19584684.pthrri(&rgZd{Q@g9vOV@rV-&@gQ@rrNrrrJrIrrQ D.#>    ###  !  M GrJrc ReZdZedeediedddddid d d  ZeZy )rz9https://download.pytorch.org/models/vgg13_bn-abd245e5.pthrri(=rg/$Q@g-V@rrg= ףp@rrNrrrJrIrrsQ G.#>    ###     M GrJrc eZdZedeediedddddid d d  Zed eedddieddddededdid ddd Z eZ y)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. )rrurrrrr) rPrQrRrrr rrrVIMAGENET1K_FEATURESrrrJrIrrs D.#>    ###  !  M " W ,7    #M"5\"5\  ! :GrJrc ReZdZedeediedddddid d d  ZeZy )rz9https://download.pytorch.org/models/vgg16_bn-6c64b313.pthrri(L?rgףp= WR@g/$V@rrgrh~@rrNrrrJrIrrrrJrc ReZdZedeediedddddid d d  ZeZy )rz6https://download.pytorch.org/models/vgg19-dcbb9e9d.pthrri(0rgMbR@gMbV@roʡ3@g rh @rrNrrrJrIrr rrJrc ReZdZedeediedddddid d d  ZeZy )rz9https://download.pytorch.org/models/vgg19_bn-c79401a0.pthrri([rgˡER@gSV@rrg/$!@rrNrrrJrIrrsQ G.#>    ###  !  M GrJr pretrained)rqT)rqrrc Jtj|}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: rlF)rverifyr|rqrrrss rIrr2**""7+G UGX 8 88rJc Jtj|}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: rlT)rrr|rs rIrrL**%%g.G T7H 7 77rJc Jtj|}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: rmF)rrr|rs rIrrfrrJc Jtj|}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: rmT)rrr|rs rIr r rrJc Jtj|}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: rnF)rrr|rs rIr!r!rrJc Jtj|}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: rnT)rrr|rs rIr"r"rrJc Jtj|}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: roF)rrr|rs rIr#r#rrJc Jtj|}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: roT)rrr|rs rIr$r$rrJ)F)6 functoolsrtypingrrrrrr rMtorch.nnr4transforms._presetsr utilsr _apirrr_metar_utilsrr__all__rSrstrrTrUr7rgrp__annotations__r|rrrrrrrrrrrrrr r!r"r#r$rrJrIrs99 5'66'B *#"))#L"T%S/*"""$ J R a p *d3U38_% %&cth{.Ct_bgj&e a  K({(K({(.K.b{(K({(, 0K0K!LM04t9h}-99WZ9_b9N90,0@0N0N!OP6:T8"238d8]`8eh8Q80, 0K0K!LM04t9h}-99WZ9_b9N90,0@0N0N!OP6:T8"238d8]`8eh8Q80, 0K0K!LM04t9h}-99WZ9_b9N90,0@0N0N!OP6:T8"238d8]`8eh8Q80, 0K0K!LM04t9h}-99WZ9_b9N90,0@0N0N!OP6:T8"238d8]`8eh8Q8rJ