U ʹj2@sddlZddlmZddlmZddlmZmZmZm Z m Z ddl Z ddl m Z ddlm mZddl mZddlmZddlmZd d lmZmZmZd d lmZd d lmZmZd ddddgZeddddgZ ee ee ede _!e Z"Gdd d e j#Z$Gddde j#Z%Gddde j#Z&Gddde j#Z'GdddeZ(eede(j)fddd d!e e(e*ee$d"d#dZ+dS)$N) namedtuple)partial)AnyCallableListOptionalTuple)Tensor)ImageClassification)_log_api_usage_once)register_modelWeights WeightsEnum)_IMAGENET_CATEGORIES)_ovewrite_named_paramhandle_legacy_interface GoogLeNetGoogLeNetOutputs_GoogLeNetOutputsGoogLeNet_Weights googlenetlogits aux_logits2 aux_logits1)rrrc seZdZddgZdeeeeeeeed e j fe e dd fd d Z e e d ddZe ee ee ee fd ddZejje e ee edddZe ed ddZZS)r aux_logitstransform_inputTFN皙?ffffff?.) num_classesrr init_weightsblocksdropout dropout_auxreturnc stt||dkr$tttg}|dkrtdt |||S| |||SdS)Nz8Scripted GoogleNet always returns GoogleNetOutputs Tuple) rwr{ryrrhjitZ is_scriptingrHrIrr})rmrur_r`Z aux_definedrrrrrsforwards     zGoogLeNet.forward)rTFNNrr )__name__ __module__ __qualname__Z __constants__intboolrrrrNModulefloatrDr rwrr{rhr~Zunusedrr}r __classcell__rrrrrprsrs.F"7c sfeZdZd eeeeeeeeedejfdd fdd Ze e e dddZ e e dd d Z Z S) rFN.) in_channelsch1x1ch3x3redch3x3ch5x5redch5x5 pool_projrnr&c st|dkrt}|||dd|_t|||dd|||ddd|_t|||dd|||ddd|_ttjddddd|||dd|_ dS)Nr r.r'r0T)r*r+r,r-) rCrDrEbranch1rNZ Sequentialbranch2branch3rObranch4) rmrrrrrrrrnrprrrsrDs    zInception.__init__rtcCs8||}||}||}||}||||g}|Sr|)rrrr)rmrurrrroutputsrrrrrsr{s      zInception._forwardcCs||}t|dSrx)r{rhrv)rmrurrrrrrsrs zInception.forward)N)rrrrrrrNrrDr rr{rrrrrrrprsrFs   rFcsJeZdZd eeeedejfeddfdd Z e e ddd Z Z S) rGNr .)rr!rnr$r&csTt|dkrt}||ddd|_tdd|_td||_tj|d|_ dS)Nr2r r.ir?r=) rCrDrEconvrNrcfc1fc2rbr$)rmrr!rnr$rprrrsrDs zInceptionAux.__init__rtcCsNt|d}||}t|d}tj||dd}||}||}|S)N)rr TZinplace) FZadaptive_avg_pool2drrhrzrelurr$rrmrurrrrrsrs     zInceptionAux.forward)Nr ) rrrrrrrNrrrDr rrrrrrrprsrGsrGcs8eZdZeeeddfdd ZeedddZZS)rEN)r out_channelskwargsr&c s8ttj||fddi||_tj|dd|_dS)NrlFgMbP?)eps)rCrDrNrgrrkbn)rmrrrrprrrsrD s zBasicConv2d.__init__rtcCs"||}||}tj|ddS)NTr)rrrrrrrrrrsrs  zBasicConv2d.forward) rrrrrrDr rrrrrrrprsrE srEc @sBeZdZedeeddddedddd d id d d ddZeZdS)rz:https://download.pytorch.org/models/googlenet-1378be20.pthr:)Z crop_sizeie)rzOhttps://github.com/pytorch/vision/tree/main/references/classification#googlenetz ImageNet-1KgoqQ@gRaV@)zacc@1zacc@5g+?g!rhH@z1These weights are ported from the original paper.)Z num_paramsZmin_size categoriesZrecipeZ_metricsZ_ops _file_sizeZ_docs)urlZ transformsmetaN) rrrrrr r IMAGENET1K_V1DEFAULTrrrrrrrsrs$ Z pretrained)weightsT)rprogress)rrrr&cKst|}|dd}|dk r`d|kr2t|ddt|ddt|ddt|dt|jdtf|}|dk r||j|dd |sd|_ d|_ d|_ n t d |S) aGoogLeNet (Inception v1) model architecture from `Going Deeper with Convolutions `_. Args: weights (:class:`~torchvision.models.GoogLeNet_Weights`, optional): The pretrained weights for the model. See :class:`~torchvision.models.GoogLeNet_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.GoogLeNet`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.GoogLeNet_Weights :members: rFNrTr"r!r)rZ check_hashz`auxiliary heads in the pretrained googlenet model are NOT pretrained, so make sure to train them)rverifygetrrKrrZload_state_dictZget_state_dictrr_r`rHrI)rrrZoriginal_aux_logitsmodelrrrrrsr-s&      ),rH collectionsr functoolsrtypingrrrrrrhZtorch.nnrNZtorch.nn.functionalZ functionalrr Ztransforms._presetsr utilsr Z_apirrr_metarZ_utilsrr__all__r__annotations__rrrrFrGrErrrrrrrrrrrss2       /#