Yh K PUddlmZddlmZmZmZmZddlZddlm Z ddl m Z ddl m Z ddlmZmZmZdd lmZdd lmZmZgd ZGd d e jZd=deeeefdede jfdZgdgdgdgddZ e!eeeeeffe"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)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_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 h/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/site-packages/torchvision/models/vgg.pyr3z 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&r6torchflattenr;)rFrKs rIforwardz VGG.forwardAsI MM!   LLOO M!Q   OOA  rJ)r$Tr%) __name__ __module__ __qualname__r4Moduleintboolfloatr3rMTensorrO __classcell__)rHs@rIrr#shk11 1031JN1`e1 111111:%,rJrFcfg batch_normr*cdg}d}|D]}|dkr|tjddgz }#tt|}tj||dd}|r.||tj|tjdgz }n||tjdgz }|}tj|S) NMr) kernel_sizestrider )r^paddingT)inplace)r4 MaxPool2drrTr>rDr9r7)rYrZlayers in_channelsvconv2ds rI make_layersrgIs FK    88 r|!<<<= =FFS! AY{A1aHHHF :62>!#4#4bgd6K6K6KLL6274#8#8#899KK =& !!rJ) @r]r]rjr]rkr]rkrkr])rhrhr]ririr]rjrjr]rkrkr]rkrkr])rhrhr]ririr]rjrjrjr]rkrkrkr]rkrkrkr])rhrhr]ririr]rjrjrjrjr]rkrkrkrkr]rkrkrkrkr])ABDEcfgsweightsprogresskwargsc &|;d|d<|jd)t|dt|jdtt t ||fi|}|*|||d|S)NFr( categoriesr')rZT)rr check_hash)metarlenrrgrpload_state_dictget_state_dict)rYrZrqrrrsmodels rI_vggr|bs!&~ < % 1 !&-W\,=W9X9X Y Y Y  DI*=== H H H HE g44hSW4XXYYY 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 `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 transformsrwN rPrQrRr rr _COMMON_META IMAGENET1K_V1DEFAULTrJrIrrusG D7.#>>>    ###        M GGGrJrc `eZdZedeediedddddid d d  ZeZd S)rz9https://download.pytorch.org/models/vgg11_bn-6002323d.pthrrijrgHzQ@gp= sV@rrgjt@rrNrrrJrIrrG G7.#>>>    ###   !    M GGGrJrc `eZdZedeediedddddid d d  ZeZd S)rz6https://download.pytorch.org/models/vgg13-19584684.pthrri(&rgZd{Q@g9vOV@rV-&@gQ@rrNrrrJrIrrG D7.#>>>    ###   !    M GGGrJrc `eZdZedeediedddddid d d  ZeZd S)rz9https://download.pytorch.org/models/vgg13_bn-abd245e5.pthrri(=rg/$Q@g-V@rrg= ףp@rrNrrrJrIrrsG G7.#>>>    ###        M GGGrJrc 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. )rrurrrrr) rPrQrRr rr rrrVIMAGENET1K_FEATURESrrrJrIrrsG D7.#>>>    ###   !    M "' W7 ,7      #M"U5\\"U5\\   !   :GGGrJrc `eZdZedeediedddddid d d  ZeZd S)rz9https://download.pytorch.org/models/vgg16_bn-6c64b313.pthrri(L?rgףp= WR@g/$V@rrgrh~@rrNrrrJrIrrrrJrc `eZdZedeediedddddid d d  ZeZd S)rz6https://download.pytorch.org/models/vgg19-dcbb9e9d.pthrri(0rgMbR@gMbV@roʡ3@g rh @rrNrrrJrIrr rrJrc `eZdZedeediedddddid d d  ZeZd S)rz9https://download.pytorch.org/models/vgg19_bn-c79401a0.pthrri([rgˡER@gSV@rrg/$!@rrNrrrJrIrrsG G7.#>>>    ###   !    M GGGrJr pretrained)rqT)rqrrc 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: rlF)rverifyr|rqrrrss rIrr23*""7++G UGX 8 8 8 88rJc 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: rlT)rrr|rs rIrrL3*%%g..G T7H 7 7 7 77rJc 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: rmF)rrr|rs rIrrfrrJc 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: rmT)rrr|rs rIrrrrJc 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: rnF)rrr|rs rIrrrrJc 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: rnT)rrr|rs rIr r rrJc 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: roF)rrr|rs rIr!r!rrJc 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: roT)rrr|rs rIr"r"rrJ)F)6 functoolsrtypingrrrrrMtorch.nnr4transforms._presetsr utilsr _apir r r_metar_utilsrr__all__rSrliststrrTrUr7rgrpdict__annotations__r|rrrrrrrrrrrrrrrr r!r"rrJrIrsE------------ 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]`8eh888QP888rJ