a PSic+@sddlZddlmZddlmZmZmZmZddlZddl m Z ddl m m Z ddlmZddlmZddlmZmZddlmZd d lmZmZd d lmZd d lmZmZd dlm Z m!Z!m"Z"gdZ#Gdddej$Z%Gdddej&Z'Gdddej(Z)Gdddej*Z+Gdddej,Z-Gdddej.Z/Gdddej0Z1Gdddej2Z3Gd d!d!eZ4ed"d#d$fd%dd&d'd(eee4efe5e5ee3d)d*d+Z6d d,lm7Z7d d-lm8Z8e7d.e4j9j:iZ;dS)/N)partial)AnyListOptionalUnion)Tensor) inception)InceptionOutputsInception_V3_Weights)ImageClassification) WeightsEnumWeights)_IMAGENET_CATEGORIES)handle_legacy_interface_ovewrite_named_param) _fuse_modules _replace_reluquantize_model)QuantizableInception3Inception_V3_QuantizedWeights inception_v3csLeZdZeeddfdd ZeedddZd eeddd d Z Z S) QuantizableBasicConv2dNargskwargsreturncs tj|i|t|_dSN)super__init__nnReLUreluselfrr __class__e/var/www/html/django/DPS/env/lib/python3.9/site-packages/torchvision/models/quantization/inception.pyr!szQuantizableBasicConv2d.__init__xrcCs"||}||}||}|Srconvbnr$)r&r,r)r)r*forwards   zQuantizableBasicConv2d.forwardis_qatrcCst|gd|dddS)Nr-T)inplace)r)r&r2r)r)r* fuse_model%sz!QuantizableBasicConv2d.fuse_model)N) __name__ __module__ __qualname__rr!rr0rboolr4 __classcell__r)r)r'r*rsrcs6eZdZeeddfdd ZeedddZZS)QuantizableInceptionANrcs&tj|dti|tj|_dSN conv_blockr r!rr" quantizedFloatFunctionalmyopr%r'r)r*r!+szQuantizableInceptionA.__init__r+cCs||}|j|dSNr_forwardr@catr&r,outputsr)r)r*r0/s zQuantizableInceptionA.forwardr5r6r7rr!rr0r9r)r)r'r*r:)sr:cs6eZdZeeddfdd ZeedddZZS)QuantizableInceptionBNrcs&tj|dti|tj|_dSr;r=r%r'r)r*r!6szQuantizableInceptionB.__init__r+cCs||}|j|dSrArBrEr)r)r*r0:s zQuantizableInceptionB.forwardrGr)r)r'r*rH4srHcs6eZdZeeddfdd ZeedddZZS)QuantizableInceptionCNrcs&tj|dti|tj|_dSr;r=r%r'r)r*r!AszQuantizableInceptionC.__init__r+cCs||}|j|dSrArBrEr)r)r*r0Es zQuantizableInceptionC.forwardrGr)r)r'r*rI?srIcs6eZdZeeddfdd ZeedddZZS)QuantizableInceptionDNrcs&tj|dti|tj|_dSr;r=r%r'r)r*r!LszQuantizableInceptionD.__init__r+cCs||}|j|dSrArBrEr)r)r*r0Ps zQuantizableInceptionD.forwardrGr)r)r'r*rJJsrJcsJeZdZeeddfdd ZeeedddZeeddd ZZ S) QuantizableInceptionENrcs>tj|dti|tj|_tj|_tj|_dSr;) r r!rr"r>r?myop1myop2myop3r%r'r)r*r!Ws  zQuantizableInceptionE.__init__r+cCs||}||}||||g}|j|d}||}||}||| |g}|j |d}t j |dddd}| |}||||g}|S)Nrr ) kernel_sizestridepadding) branch1x1 branch3x3_1 branch3x3_2a branch3x3_2brLrDbranch3x3dbl_1branch3x3dbl_2branch3x3dbl_3abranch3x3dbl_3brMF avg_pool2d branch_pool)r&r,rR branch3x3 branch3x3dblr\rFr)r)r*rC]s      zQuantizableInceptionE._forwardcCs||}|j|dSrA)rCrNrDrEr)r)r*r0rs zQuantizableInceptionE.forward) r5r6r7rr!rrrCr0r9r)r)r'r*rKUsrKcs&eZdZeeddfdd ZZS)QuantizableInceptionAuxNrcstj|dti|dSr;)r r!rr%r'r)r*r!ysz QuantizableInceptionAux.__init__)r5r6r7rr!r9r)r)r'r*r_wsr_csPeZdZdeeeddfdd Zeedd d Zde edd d d Z Z S)rTFN) num_classes aux_logitstransform_inputrc sBtj|||tttttttgdt j j |_ t j j |_dS)N)rarbrcinception_blocks)r r!rr:rHrIrJrKr_torchao quantization QuantStubquant DeQuantStubdequant)r&rarbrcr'r)r*r!~szQuantizableInception3.__init__r+cCsj||}||}||\}}||}|jo6|j}tjrZ|sPt dt ||S| ||SdS)NzIScripted QuantizableInception3 always returns QuantizableInception3 Tuple) _transform_inputrirCrktrainingrbrejit is_scriptingwarningswarnr eager_outputs)r&r,aux aux_definedr)r)r*r0s       zQuantizableInception3.forwardr1cCs(|D]}t|tur||qdS)aFuse conv/bn/relu modules in inception model Fuse conv+bn+relu/ conv+relu/conv+bn modules to prepare for quantization. Model is modified in place. Note that this operation does not change numerics and the model after modification is in floating point N)modulestyperr4)r&r2mr)r)r*r4s  z QuantizableInception3.fuse_model)r`TF)N) r5r6r7intr8r!rr r0rr4r9r)r)r'r*r}s rc @sFeZdZedeedddddeddejd d d d id ddZ e Z dS)rzUhttps://download.pytorch.org/models/quantized/inception_v3_google_fbgemm-71447a44.pthi+iV) crop_size resize_sizeir)Kr{fbgemmzdhttps://github.com/pytorch/vision/tree/main/references/classification#post-training-quantized-modelsz ImageNet-1Kg%CKS@g-VW@)zacc@1zacc@5z These weights were produced by doing Post Training Quantization (eager mode) on top of the unquantized weights listed below. ) num_paramsmin_size categoriesbackendrecipe unquantized_metrics_docs)url transformsmetaN) r5r6r7rrr rr IMAGENET1K_V1IMAGENET1K_FBGEMM_V1DEFAULTr)r)r)r*rs$ r pretrainedcCs|ddrtjStjS)NquantizeF)getrrr r)rr)r)r*s r)weightsTF)rprogressr)rrrrrcKs|rtnt|}|dd}|durxd|vr:t|ddt|ddt|dt|jdd|jvrxt|d|jd|dd }tfi|}t ||rt |||dur|r|sd|_ d|_ | |j|d |s|sd|_ d|_ |S) aInception v3 model architecture from `Rethinking the Inception Architecture for Computer Vision `__. .. note:: **Important**: In contrast to the other models the inception_v3 expects tensors with a size of N x 3 x 299 x 299, so ensure your images are sized accordingly. .. note:: Note that ``quantize = True`` returns a quantized model with 8 bit weights. Quantized models only support inference and run on CPUs. GPU inference is not yet supported. Args: weights (:class:`~torchvision.models.quantization.Inception_V3_QuantizedWeights` or :class:`~torchvision.models.Inception_V3_Weights`, optional): The pretrained weights for the model. See :class:`~torchvision.models.quantization.Inception_V3_QuantizedWeights` 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. quantize (bool, optional): If True, return a quantized version of the model. Default is False. **kwargs: parameters passed to the ``torchvision.models.quantization.QuantizableInception3`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.quantization.Inception_V3_QuantizedWeights :members: .. autoclass:: torchvision.models.Inception_V3_Weights :members: :noindex: rbFNrcTrarrr|)r)rr verifyrrlenrpoprrrrb AuxLogitsload_state_dictget_state_dict)rrrroriginal_aux_logitsrmodelr)r)r*rs.1      r) _ModelURLs) model_urlsZinception_v3_google_fbgemm)sX          "2 G