a PSic"@s~ddlmZddlmZmZddlZddlmZddlmm Z ddl m Z ddl m Z ddlmZmZdd lmZdd lmZmZgd ZGd d d ejZGdddejZeeeeeedddZedddZGdddeZGdddeZedej fddddeeeeeddd Z!edej fddddeeeeedd!d"Z"dd#lm#Z#e#ej j$ej j$d$Z%dS)%)partial)AnyOptionalN)ImageClassification)_log_api_usage_once) WeightsEnumWeights)_IMAGENET_CATEGORIES)handle_legacy_interface_ovewrite_named_param) SqueezeNetSqueezeNet1_0_WeightsSqueezeNet1_1_Weights squeezenet1_0 squeezenet1_1cs>eZdZeeeeddfdd ZejejdddZZS)FireN)inplanessqueeze_planesexpand1x1_planesexpand3x3_planesreturncsvt||_tj||dd|_tjdd|_tj||dd|_tjdd|_ tj||ddd|_ tjdd|_ dS)Nr kernel_sizeTinplace)rpadding) super__init__rnnConv2dsqueezeReLUsqueeze_activation expand1x1expand1x1_activation expand3x3expand3x3_activation)selfrrrr __class__Y/var/www/html/django/DPS/env/lib/python3.9/site-packages/torchvision/models/squeezenet.pyr s z Fire.__init__xrcCs8|||}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 rcs>eZdZd eeeddfdd Zejejdd d Z Z S) r1_0?N)version num_classesdropoutrcs:tt|||_|dkrttjdddddtjddtjdddd 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 |_ n|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]X}t|tjr||ur tj|jdddn t|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)rr rr@r! Sequentialr"r$ MaxPool2drfeatures ValueErrorDropoutAdaptiveAvgPool2d classifiermodules isinstanceinitnormal_weightkaiming_uniform_bias constant_)r*r?r@rAZ final_convmr+r-r.r %s^                        zSqueezeNet.__init__r/cCs ||}||}t|dSr1)rSrWr2flattenr4r-r-r.r5^s  zSqueezeNet.forward)r<r=r>) r6r7r8strr9floatr r2r:r5r;r-r-r+r.r$s9r)r?weightsprogresskwargsrcKsL|durt|dt|jdt|fi|}|durH||j|d|S)Nr@ categories)re)r lenmetarload_state_dictget_state_dict)r?rdrerfmodelr-r-r. _squeezenetds rmz@https://github.com/pytorch/vision/pull/49#issuecomment-277560717zXThese weights reproduce closely the results of the paper using a simple training recipe.)rgrecipe_docsc @s@eZdZedeeddiedddddd id d ZeZd S) rz>https://download.pytorch.org/models/squeezenet1_0-b66bff10.pth crop_size)rsi  ImageNet-1Kg M@g{GT@zacc@1zacc@5min_size num_params_metricsurl transformsriN r6r7r8r rr _COMMON_META IMAGENET1K_V1DEFAULTr-r-r-r.r|s  rc @s@eZdZedeeddiedddddd id d ZeZd S) rz>https://download.pytorch.org/models/squeezenet1_1-b8a52dc0.pthrprq)ri(rtgX9M@g-'T@rurvrzNr}r-r-r-r.rs  r pretrained)rdT)rdre)rdrerfrcKst|}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<)rverifyrmrdrerfr-r-r.rs rcKst|}td||fi|S)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)rrrmrr-r-r.rs r) _ModelURLs)rr)& functoolsrtypingrrr2torch.nnr!Z torch.nn.initrZZtransforms._presetsrutilsr_apir r _metar _utilsr r __all__Modulerrrbboolrmr~rrrrrrr{ model_urlsr-r-r-r.sT     A