U (d"@sddlmZddlmZmZddlZddlmZddlmm Z ddl m Z ddl m Z ddlmZmZdd lmZdd lmZmZd d d ddgZGdddejZGdd d ejZeeeeeedddZedddZGdd d eZGdd d eZedej fddddeeeeedd dZ!edej fddddeeeeedd!dZ"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_sizeTZinplace)rpadding) super__init__rnnConv2dsqueezeReLUsqueeze_activation expand1x1expand1x1_activation expand3x3expand3x3_activation)selfrrrr __class__A/tmp/pip-unpacked-wheel-vx7f76es/torchvision/models/squeezenet.pyrs z Fire.__init__xrcCs8|||}t||||||gdSNr)r$r"torchcatr&r%r(r'r)r/r,r,r-forwards z Fire.forward) __name__ __module__ __qualname__intrr1Tensorr4 __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)rrCZ ceil_mode@ 0ii1_1zUnsupported SqueezeNet version z: 1_0 or 1_1 expectedrr)p)rrgg{Gz?)ZmeanZstdr)rrrr?r Z Sequentialr!r#Z MaxPool2drfeatures ValueErrorZDropoutZAdaptiveAvgPool2d classifiermodules isinstanceinitZnormal_ZweightZkaiming_uniform_ZbiasZ constant_)r)r>r?r@Z final_convmr*r,r-r%sd                          zSqueezeNet.__init__r.cCs ||}||}t|dSr0)rMrOr1flattenr3r,r,r-r4^s  zSqueezeNet.forward)r;r<r=) r5r6r7strr8floatrr1r9r4r:r,r,r*r-r$s9)r>weightsprogresskwargsrcKsH|dk rt|dt|jdt|f|}|dk rD||j|d|S)Nr? categories)rX)r lenmetarZload_state_dictZget_state_dict)r>rWrXrYmodelr,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.)rZZrecipeZ_docsc @s<eZdZedeeddedddddd id d ZeZd S) rz>https://download.pytorch.org/models/squeezenet1_0-b66bff10.pthZ crop_size)rai  ImageNet-1Kg M@g{GT@zacc@1zacc@5Zmin_sizeZ num_paramsZ_metricsurlZ transformsr\N r5r6r7r rr _COMMON_META IMAGENET1K_V1DEFAULTr,r,r,r-r|s c @s<eZdZedeeddedddddd id d ZeZd S) rz>https://download.pytorch.org/models/squeezenet1_1-b8a52dc0.pthr_r`)rki(rbgX9M@g-'T@rcrdreNrgr,r,r,r-rs Z pretrained)rWT)rWrX)rWrXrYrcKst|}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^rWrXrYr,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: rK)rrlr^rmr,r,r-rs ) _ModelURLs)rr)& functoolsrtypingrrr1Ztorch.nnr Z torch.nn.initrRZtransforms._presetsrutilsrZ_apir r Z_metar _utilsr r __all__ModulerrrUboolr^rhrrrirrrnrfZ model_urlsr,r,r,r-s`     A