U h;"@srddlmZddlmZmZddlZddlmZddlmm Z ddl m Z ddl m Z ddlmZmZmZdd lmZdd lmZmZd d d ddgZGdddejZGdd d ejZeeeeeedddZedddZGdd d eZGdd d eZ eedej!fddddeeeeedd dZ"eede j!fddddee eeedd!dZ#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_1cs>eZdZeeeeddfdd ZejejdddZZS)FireN)inplanessqueeze_planesexpand1x1_planesexpand3x3_planesreturncsvt||_tj||dd|_tjdd|_tj||dd|_tjdd|_ tj||ddd|_ tjdd|_ dS)Nr kernel_sizeTZinplace)rpadding) super__init__rnnConv2dsqueezeReLUsqueeze_activation expand1x1expand1x1_activation expand3x3expand3x3_activation)selfrrrr __class__O/var/www/html/venv/lib/python3.8/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||kr tj|jdddn t|j|jdk rt|jdqdS)Nr<r`r)rstrideTr)rrDZ ceil_mode@ 0ii1_1zUnsupported SqueezeNet version z: 1_0 or 1_1 expectedrr)p)rrgg{Gz?)ZmeanZstdr)rr rr@r!Z Sequentialr"r$Z MaxPool2drfeatures ValueErrorZDropoutZAdaptiveAvgPool2d classifiermodules isinstanceinitZnormal_weightZkaiming_uniform_ZbiasZ constant_)r*r?r@rAZ final_convmr+r-r.r %sd                          zSqueezeNet.__init__r/cCs ||}||}t|dSr1)rNrPr2flattenr4r-r-r.r5^s  zSqueezeNet.forward)r<r=r>) r6r7r8strr9floatr r2r:r5r;r-r-r+r.r$s9)r?weightsprogresskwargsrcKsJ|dk rt|dt|jdt|f|}|dk rF||j|dd|S)Nr@ categoriesT)rZZ check_hash)r lenmetarZload_state_dictZget_state_dict)r?rYrZr[modelr-r-r. _squeezenetds  r`z@https://github.com/pytorch/vision/pull/49#issuecomment-277560717zXThese weights reproduce closely the results of the paper using a simple training recipe.)r\ZrecipeZ_docsc @s@eZdZedeeddedddddd id d d d ZeZdS)rz>https://download.pytorch.org/models/squeezenet1_0-b66bff10.pthZ crop_size)rci  ImageNet-1Kg M@g{GT@zacc@1zacc@5gh|?5?g&1@Zmin_sizeZ num_paramsZ_metricsZ_ops _file_sizeurlZ transformsr^N r6r7r8r rr _COMMON_META IMAGENET1K_V1DEFAULTr-r-r-r.r|s  c @s@eZdZedeeddedddddd id d d d ZeZdS)rz>https://download.pytorch.org/models/squeezenet1_1-b8a52dc0.pthrarb)rni(rdgX9M@g-'T@regtV?g"~@rfrhNrjr-r-r-r.rs  Z pretrained)rYT)rYrZ)rYrZr[rcKst|}td||f|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<)rverifyr`rYrZr[r-r-r.rs cKst|}td||f|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: rL)rror`rpr-r-r.rs )$ functoolsrtypingrrr2Ztorch.nnr!Z torch.nn.initrSZtransforms._presetsrutilsrZ_apir r r _metar Z_utilsr r__all__ModulerrrWboolr`rkrrrlrrr-r-r-r.sX     A