o V h;" @sddlmZddlmZmZddlZddlmZddlmm Z ddl m Z ddl m Z ddlmZmZmZdd lmZdd lmZmZgd ZGd d d ejZGdddejZdedeedededef ddZedddZGdddeZGdddeZ eedej!fddd d!deedededefd"d#Z"eede j!fddd d!dee dededefd$d%Z#dS)&)partial)AnyOptionalN)ImageClassification)_log_api_usage_once)register_modelWeights WeightsEnum)_IMAGENET_CATEGORIES)_ovewrite_named_paramhandle_legacy_interface) SqueezeNetSqueezeNet1_0_WeightsSqueezeNet1_1_Weights squeezenet1_0 squeezenet1_1c sHeZdZdededededdf fdd Zd ejdejfd d ZZS) Fireinplanessqueeze_planesexpand1x1_planesexpand3x3_planesreturnNcsvt||_tj||dd|_tjdd|_tj||dd|_tjdd|_ tj||ddd|_ tjdd|_ dS)Nr kernel_sizeTinplace)rpadding) super__init__rnnConv2dsqueezeReLUsqueeze_activation expand1x1expand1x1_activation expand3x3expand3x3_activation)selfrrrr __class__Q/var/www/vscode/kcb/lib/python3.10/site-packages/torchvision/models/squeezenet.pyr!s z Fire.__init__xcCs8|||}t||||||gdSNr)r&r$torchcatr(r'r*r)r+r0r.r.r/forwards z Fire.forward) __name__ __module__ __qualname__intr!r2Tensorr5 __classcell__r.r.r,r/rs" rc sFeZdZddedededdffd d Zd ejdejfd d Z Z S)r1_0?version num_classesdropoutrNcs6tt|||_|dkrhttjdddddtjddtjdddd t dd d d t d d d d t d d d d tjdddd t dd d d t ddddt ddddt dd ddtjdddd t dd dd |_ nd|dkrttjdd dddtjddtjdddd t d d d d t d d d d tjdddd t d d d d t dd d d tjdddd t ddddt ddddt dd ddt dd dd |_ nt d|dtjd|jdd}ttj |d|tjddt d|_|D]+}t|tjr||urtj|jdddnt|j|jdurt|jdqdS)Nr<r`r)rstrideTr)rrD ceil_mode@ 0ii1_1zUnsupported SqueezeNet version z: 1_0 or 1_1 expectedrr)p)rrgg{Gz?)meanstdr)r r!rr@r" Sequentialr#r% MaxPool2drfeatures ValueErrorDropoutAdaptiveAvgPool2d classifiermodules isinstanceinitnormal_weightkaiming_uniform_bias constant_)r+r?r@rA final_convmr,r.r/r!%sb                       zSqueezeNet.__init__r0cCs ||}||}t|dSr1)rSrWr2flattenr4r.r.r/r5^s   zSqueezeNet.forward)r<r=r>) r6r7r8strr9floatr!r2r:r5r;r.r.r,r/r$s 9rr?weightsprogresskwargsrcKsN|durt|dt|jdt|fi|}|dur%||j|dd|S)Nr@ categoriesT)rf check_hash)r lenmetarload_state_dictget_state_dict)r?rerfrgmodelr.r.r/ _squeezenetds roz@https://github.com/pytorch/vision/pull/49#issuecomment-277560717zXThese weights reproduce closely the results of the paper using a simple training recipe.)rhrecipe_docsc @DeZdZedeeddiedddddd id d d d ZeZdS)rz>https://download.pytorch.org/models/squeezenet1_0-b66bff10.pth crop_size)rvi  ImageNet-1Kg M@g{GT@zacc@1zacc@5gh|?5?g&1@min_size num_params_metrics_ops _file_sizeurl transformsrkN r6r7r8r rr _COMMON_META IMAGENET1K_V1DEFAULTr.r.r.r/r|& rc @rr)rz>https://download.pytorch.org/models/squeezenet1_1-b8a52dc0.pthrsrt)ri(rwgX9M@g-'T@rxgtV?g"~@ryrNrr.r.r.r/rrr pretrained)reT)rerfcKt|}td||fi|S)aSqueezeNet model architecture from the `SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <0.5MB model size `_ paper. Args: weights (:class:`~torchvision.models.SqueezeNet1_0_Weights`, optional): The pretrained weights to use. See :class:`~torchvision.models.SqueezeNet1_0_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.squeezenet.SqueezeNet`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.SqueezeNet1_0_Weights :members: r<)rverifyrorerfrgr.r.r/rs rcKr)a/SqueezeNet 1.1 model from the `official SqueezeNet repo `_. SqueezeNet 1.1 has 2.4x less computation and slightly fewer parameters than SqueezeNet 1.0, without sacrificing accuracy. Args: weights (:class:`~torchvision.models.SqueezeNet1_1_Weights`, optional): The pretrained weights to use. See :class:`~torchvision.models.SqueezeNet1_1_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.squeezenet.SqueezeNet`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.SqueezeNet1_1_Weights :members: rM)rrrorr.r.r/rs r)$ functoolsrtypingrrr2torch.nnr" torch.nn.initrZtransforms._presetsrutilsr_apir r r _metar _utilsr r__all__Modulerrrcboolrorrrrrrr.r.r.r/sl     @