U ʹjF @sddlmZddlmZmZmZmZmZddlZddl m Z ddlm Z ddl m Z mZmZmZmZmZmZddlmZdd lmZmZmZdd lmZdd lmZmZd d lm Z m!Z!m"Z"dddddddddg Z#Gddde Z$GdddeZ%GdddeZ&eee$e%fee'eee(e(ee&dddZ)ded d!d"d#Z*Gd$ddeZ+Gd%ddeZ,Gd&ddeZ-Gd'ddeZ.ed(d)ed*d+d,fd-dd.d/d0eee+efe(e(ee&d1d2dZ/ed3d)ed*d4d,fd-dd.d/d0eee,efe(e(ee&d1d5dZ0ed6d)ed*d7d,fd-dd.d/d0eee-efe(e(ee&d1d8dZ1ed9d)ed*d:d,fd-dd.d/d0eee.efe(e(ee&d1d;dZ2dS)<)partial)AnyListOptionalTypeUnionN)Tensor) BasicBlock BottleneckResNetResNet18_WeightsResNet50_WeightsResNeXt101_32X8D_WeightsResNeXt101_64X4D_Weights)ImageClassification)register_modelWeights WeightsEnum)_IMAGENET_CATEGORIES)_ovewrite_named_paramhandle_legacy_interface) _fuse_modules _replace_reluquantize_modelQuantizableResNetResNet18_QuantizedWeightsResNet50_QuantizedWeights!ResNeXt101_32X8D_QuantizedWeights!ResNeXt101_64X4D_QuantizedWeightsresnet18resnet50resnext101_32x8dresnext101_64x4dcsLeZdZeeddfdd ZeedddZd eeddd d Z Z S) QuantizableBasicBlockNargskwargsreturncs tj||tjj|_dSN)super__init__torchnn quantizedFloatFunctionaladd_reluselfr(r) __class__e/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torchvision/models/quantization/resnet.pyr-&szQuantizableBasicBlock.__init__xr*cCs\|}||}||}||}||}||}|jdk rJ||}|j||}|Sr+)conv1bn1reluconv2bn2 downsampler2r4r:identityoutr7r7r8forward*s       zQuantizableBasicBlock.forwardis_qatr*cCs>t|dddgddgg|dd|jr:t|jdd g|dddS) Nr;r<r=r>r?TZinplace01rr@r4rFr7r7r8 fuse_model;sz QuantizableBasicBlock.fuse_model)N __name__ __module__ __qualname__rr-rrDrboolrL __classcell__r7r7r5r8r&%sr&csLeZdZeeddfdd ZeedddZd eeddd d Z Z S) QuantizableBottleneckNr'cs:tj||tj|_tjdd|_tjdd|_dS)NFrG) r,r-r/r0r1 skip_add_reluZReLUrelu1relu2r3r5r7r8r-Bs zQuantizableBottleneck.__init__r9cCsz|}||}||}||}||}||}||}||}||}|jdk rh||}|j ||}|Sr+) r;r<rUr>r?rVconv3bn3r@rTr2rAr7r7r8rDHs          zQuantizableBottleneck.forwardrEcCsFt|dddgdddgddgg|d d |jrBt|jd d g|d d dS) Nr;r<rUr>r?rVrWrXTrGrHrIrJrKr7r7r8rLZsz QuantizableBottleneck.fuse_model)NrMr7r7r5r8rSAsrScsLeZdZeeddfdd ZeedddZd eeddd d Z Z S) rNr'cs.tj||tjj|_tjj|_dSr+) r,r-r.ZaoZ quantizationZ QuantStubquantZ DeQuantStubdequantr3r5r7r8r-cszQuantizableResNet.__init__r9cCs"||}||}||}|Sr+)rYZ _forward_implrZ)r4r:r7r7r8rDis   zQuantizableResNet.forwardrEcCsJt|dddg|dd|D]&}t|tks:t|tkr||qdS)aFuse conv/bn/relu modules in resnet models 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 r;r<r=TrGN)rmodulestyperSr&rL)r4rFmr7r7r8rLrs zQuantizableResNet.fuse_model)NrMr7r7r5r8rbs )blocklayersweightsprogressquantizer)r*cKs|dk r:t|dt|jdd|jkr:t|d|jd|dd}t||f|}t||rjt|||dk r||j|dd|S)NZ num_classes categoriesbackendfbgemmT)raZ check_hash) rlenmetapoprrrZload_state_dictZget_state_dict)r^r_r`rarbr)rdmodelr7r7r8_resnets   rj)rrrezdhttps://github.com/pytorch/vision/tree/main/references/classification#post-training-quantized-modelsz These weights were produced by doing Post Training Quantization (eager mode) on top of the unquantized weights listed below. )Zmin_sizercrdrecipeZ_docsc @sBeZdZedeeddedejddddid d d d Z e Z d S)rzJhttps://download.pytorch.org/models/quantized/resnet18_fbgemm_16fa66dd.pth crop_sizei(^ ImageNet-1KgV-_Q@g r8V@zacc@1zacc@5g/$?g`"y&@ num_params unquantized_metrics_ops _file_sizeurlZ transformsrgN) rNrOrPrrr _COMMON_METAr IMAGENET1K_V1IMAGENET1K_FBGEMM_V1DEFAULTr7r7r7r8rs  c @sveZdZedeeddedejddddid d d d Z ed eedddedej ddddid dd d Z e Z dS)rzJhttps://download.pytorch.org/models/quantized/resnet50_fbgemm_bf931d71.pthrlrmi(rog{GR@gjt4W@rpgB`"[@gM8@rqrwzJhttps://download.pytorch.org/models/quantized/resnet50_fbgemm-23753f79.pthrnZ resize_sizeg5^I T@gX9vW@g8@N) rNrOrPrrrryr rzr{ IMAGENET1K_V2IMAGENET1K_FBGEMM_V2r|r7r7r7r8rs>  c @sveZdZedeeddedejddddid d d d Z ed eedddedej ddddid dd d Z e Z dS)r zQhttps://download.pytorch.org/models/quantized/resnext101_32x8_fbgemm_09835ccf.pthrlrmi(JrogvS@gQW@rpgDli0@gV-U@rqrwzQhttps://download.pytorch.org/models/quantized/resnext101_32x8_fbgemm-ee16d00c.pthr}r~g~jT@g rX@gzGU@N) rNrOrPrrrryrrzr{rrr|r7r7r7r8r s>  c @sFeZdZedeedddeddejddd d id d d dZ e Z dS)r!zRhttps://download.pytorch.org/models/quantized/resnext101_64x4d_fbgemm-605a1cb3.pthrlr}r~i(mz+https://github.com/pytorch/vision/pull/5935rogxT@g/X@rpgQ.@g$cT@)rrrkrsrtrurvrwN) rNrOrPrrrryrrzr{r|r7r7r7r8r!s" Zquantized_resnet18)nameZ pretrainedcCs|ddrtjStjSNrbF)getrr{r rzr)r7r7r8s r)r`TF)r`rarb)r`rarbr)r*cKs.|rtnt|}ttddddg|||f|S)aResNet-18 model from `Deep Residual Learning for Image Recognition `_ .. 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.ResNet18_QuantizedWeights` or :class:`~torchvision.models.ResNet18_Weights`, optional): The pretrained weights for the model. See :class:`~torchvision.models.quantization.ResNet18_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.QuantizableResNet`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.quantization.ResNet18_QuantizedWeights :members: .. autoclass:: torchvision.models.ResNet18_Weights :members: :noindex: r)rr verifyrjr&r`rarbr)r7r7r8r"s-Zquantized_resnet50cCs|ddrtjStjSr)rrr{r rzrr7r7r8rQs cKs.|rtnt|}ttddddg|||f|S)aResNet-50 model from `Deep Residual Learning for Image Recognition `_ .. 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.ResNet50_QuantizedWeights` or :class:`~torchvision.models.ResNet50_Weights`, optional): The pretrained weights for the model. See :class:`~torchvision.models.quantization.ResNet50_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.QuantizableResNet`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.quantization.ResNet50_QuantizedWeights :members: .. autoclass:: torchvision.models.ResNet50_Weights :members: :noindex: r)rr rrjrSrr7r7r8r#Ms-Zquantized_resnext101_32x8dcCs|ddrtjStjSr)rr r{rrzrr7r7r8rs cKsF|rtnt|}t|ddt|ddttddddg|||f|S)aResNeXt-101 32x8d model from `Aggregated Residual Transformation for Deep Neural Networks `_ .. 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.ResNeXt101_32X8D_QuantizedWeights` or :class:`~torchvision.models.ResNeXt101_32X8D_Weights`, optional): The pretrained weights for the model. See :class:`~torchvision.models.quantization.ResNet101_32X8D_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.QuantizableResNet`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.quantization.ResNeXt101_32X8D_QuantizedWeights :members: .. autoclass:: torchvision.models.ResNeXt101_32X8D_Weights :members: :noindex: groups width_per_grouprr)r rrrrjrSrr7r7r8r$s-  Zquantized_resnext101_64x4dcCs|ddrtjStjSr)rr!r{rrzrr7r7r8rs cKsF|rtnt|}t|ddt|ddttddddg|||f|S)aResNeXt-101 64x4d model from `Aggregated Residual Transformation for Deep Neural Networks `_ .. 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.ResNeXt101_64X4D_QuantizedWeights` or :class:`~torchvision.models.ResNeXt101_64X4D_Weights`, optional): The pretrained weights for the model. See :class:`~torchvision.models.quantization.ResNet101_64X4D_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.QuantizableResNet`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.quantization.ResNeXt101_64X4D_QuantizedWeights :members: .. autoclass:: torchvision.models.ResNeXt101_64X4D_Weights :members: :noindex: r@rrrr)r!rrrrjrSrr7r7r8r%s-  )3 functoolsrtypingrrrrrr.Ztorch.nnr/rZtorchvision.models.resnetr r r r r rrZtransforms._presetsrZ_apirrr_metarZ_utilsrrutilsrrr__all__r&rSrintrQrjryrrr r!r"r#r$r%r7r7r7r8s   $   !  && ) ) +