:j K hUddlmZddlmZmZmZmZddlZddlm Z ddl m Z ddl m Z ddlmZmZmZdd lmZdd lmZmZgd ZGd d e j0Zd>2GG%%ahh2GG%%affa0299-GGOOAHHa6GG%%affa0 1 xc|j|}|j|}tj|d}|j |}|S)Nr )r$r4torchflattenr9)rDrIs rGforwardz VGG.forwardAs@ MM!  LLO MM!Q  OOA rH)iTg?) __name__ __module__ __qualname__r2Moduleintboolfloatr1rKTensorrM __classcell__)rFs@rGrr#sRhk1 1031JN1`e1 1:%,,rHrcfg 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)r2 MaxPool2drrRr<rBr7r5)rWrXlayers in_channelsvconv2ds rG make_layersreIs FK   8 r||!<= =FS! AYY{A1aHF62>>!#4bggd6KLL62774#899K  ==& !!rH) @r[r[rhr[rir[ririr[)rfrfr[rgrgr[rhrhr[ririr[ririr[)rfrfr[rgrgr[rhrhrhr[riririr[riririr[)rfrfr[rgrgr[rhrhrhrhr[ririririr[ririririr[)ABDEcfgsweightsprogresskwargsc |7d|d<|jd#t|dt|jdtt t ||fi|}|"|j |j|d|S)NFr& categoriesr%)rXT)rp check_hash)metarlenrrernload_state_dictget_state_dict)rWrXrorprqmodels rG_vggrzbs!&~ << % 1 !&-W\\,=W9X Y  DI*= H HE g44hSW4XY LrH) 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_sizersrecipe_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 transformsruN rNrOrPr rr _COMMON_META IMAGENET1K_V1DEFAULTrHrGrrusQ D.#>    ###     M GrHrc ReZdZedeediedddddid d d  ZeZy )rz9https://download.pytorch.org/models/vgg11_bn-6002323d.pthrrijrgHzQ@gp= sV@rrgjt@rrNrrrHrGrrQ G.#>    ###  !  M GrHrc ReZdZedeediedddddid d d  ZeZy )rz6https://download.pytorch.org/models/vgg13-19584684.pthrri(&rgZd{Q@g9vOV@rV-&@gQ@rrNrrrHrGrrQ D.#>    ###  !  M GrHrc ReZdZedeediedddddid d d  ZeZy )rz9https://download.pytorch.org/models/vgg13_bn-abd245e5.pthrri(=rg/$Q@g-V@rrg= ףp@rrNrrrHrGrrsQ G.#>    ###     M GrHrc 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. )rrsr}rrrr~) rNrOrPr rr rrrTIMAGENET1K_FEATURESrrrHrGrrs D.#>    ###  !  M " W ,7    #M"5\"5\  ! :GrHrc ReZdZedeediedddddid d d  ZeZy )rz9https://download.pytorch.org/models/vgg16_bn-6c64b313.pthrri(L?rgףp= WR@g/$V@rrgrh~@rrNrrrHrGrrrrHrc ReZdZedeediedddddid d d  ZeZy )rz6https://download.pytorch.org/models/vgg19-dcbb9e9d.pthrri(0rgMbR@gMbV@roʡ3@g rh @rrNrrrHrGrr rrHrc ReZdZedeediedddddid d d  ZeZy )rz9https://download.pytorch.org/models/vgg19_bn-c79401a0.pthrri([rgˡER@gSV@rrg/$!@rrNrrrHrGrrsQ G.#>    ###  !  M GrHr pretrained)roT)rorpc 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: rjF)rverifyrzrorprqs rGrr2**""7+G UGX 8 88rHc 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: rjT)rrrzrs rGrrL**%%g.G T7H 7 77rHc 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: rkF)rrrzrs rGrrfrrHc 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: rkT)rrrzrs rGrrrrHc 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: rlF)rrrzrs rGrrrrHc 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: rlT)rrrzrs rGr r rrHc 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: rmF)rrrzrs rGr!r!rrHc 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: rmT)rrrzrs rGr"r"rrH)F)6 functoolsrtypingrrrrrKtorch.nnr2transforms._presetsr utilsr _apir r r_metar_utilsrr__all__rQrliststrrRrSr5rerndict__annotations__rzrrrrrrrrrrrrrrrr r!r"rrHrGrs-- 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]`8eh8Q8rH