j#ddlmZddlmZmZddlmZddlmZddl m Z m Z m Z ddl mZdd lmZmZmZdd lmZmZmZmZmZd d lmZgd ZGddeZGddejZedddZGdde ZGdde Z dede!dee"defdZ#e edej$fdej%f d!d"d!d!ej%d#d$eed%e"dee!d&ee"d'eed(edefd)Z&e ede j$fdej%f d!d"d!d!ej%d#d$ee d%e"dee!d&ee"d'eed(edefd*Z'd!S)+)partial)AnyOptional)nn)SemanticSegmentation)register_modelWeights WeightsEnum)_VOC_CATEGORIES)_ovewrite_value_paramhandle_legacy_interfaceIntermediateLayerGetter)ResNet resnet101ResNet101_Weightsresnet50ResNet50_Weights)_SimpleSegmentationModel)FCNFCN_ResNet50_WeightsFCN_ResNet101_Weights fcn_resnet50 fcn_resnet101ceZdZdZdS)ra Implements FCN model from `"Fully Convolutional Networks for Semantic Segmentation" `_. Args: backbone (nn.Module): the network used to compute the features for the model. The backbone should return an OrderedDict[Tensor], with the key being "out" for the last feature map used, and "aux" if an auxiliary classifier is used. classifier (nn.Module): module that takes the "out" element returned from the backbone and returns a dense prediction. aux_classifier (nn.Module, optional): auxiliary classifier used during training N)__name__ __module__ __qualname____doc__TC:\Projects\Historical\venv\Lib\site-packages\torchvision/models/segmentation/fcn.pyrrs   Dr#rc,eZdZdededdffd ZxZS)FCNHead in_channelschannelsreturnNc  |dz}tj||dddtj|tjtjdtj||dg}t j|dS)NrrF)paddingbiasg?)rConv2d BatchNorm2dReLUDropoutsuper__init__)selfr'r(inter_channelslayers __class__s r$r3zFCNHead.__init__%sz$) Ik>1ae L L L N> * * GII JsOO Inh 2 2   &!!!!r#)rrr intr3 __classcell__)r7s@r$r&r&$sR "C "3 "4 " " " " " " " " " "r#r&)rrz These weights were trained on a subset of COCO, using only the 20 categories that are present in the Pascal VOC dataset. ) categoriesmin_size_docsc beZdZedeediedddddd id d d  ZeZdS)rzBhttps://download.pytorch.org/models/fcn_resnet50_coco-1167a1af.pth resize_sizeijzPhttps://github.com/pytorch/vision/tree/main/references/segmentation#fcn_resnet50COCO-val2017-VOC-labelsg@N@gV@miou pixel_accgmc@g?5^I`@ num_paramsrecipe_metrics_ops _file_sizeurl transformsmetaN rrr r rr _COMMON_METACOCO_WITH_VOC_LABELS_V1DEFAULTr"r#r$rr<s%g P7/SAAA    "h) !%,, !    "&GGGr#rc beZdZedeediedddddd id d d  ZeZdS)rzChttps://download.pytorch.org/models/fcn_resnet101_coco-7ecb50ca.pthr>r?ij<zWhttps://github.com/pytorch/vision/tree/main/references/segmentation#deeplabv3_resnet101rAgO@gV@rBgV-m@gni@rErKNrOr"r#r$rrQs%g Q7/SAAA    "o) !%,, !    "&GGGr#rbackbone num_classesauxr)cddi}|rd|d<t||}|rtd|nd}td|}t|||S)Nlayer4outrVlayer3) return_layersii)rr&r)rTrUrVr[aux_classifier classifiers r$ _fcn_resnetr^fsl u%M ("' h&x}MMMH36@WT;///DN{++J x^ 4 44r# pretrainedpretrained_backbone)weightsweights_backboneNT)raprogressrUaux_lossrbrarcrdrbkwargsc t|}tj|}|=d}td|t |jd}td|d}n|d}t |gd}t|||}|*|| |d |S) a\Fully-Convolutional Network model with a ResNet-50 backbone from the `Fully Convolutional Networks for Semantic Segmentation `_ paper. .. betastatus:: segmentation module Args: weights (:class:`~torchvision.models.segmentation.FCN_ResNet50_Weights`, optional): The pretrained weights to use. See :class:`~torchvision.models.segmentation.FCN_ResNet50_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. num_classes (int, optional): number of output classes of the model (including the background). aux_loss (bool, optional): If True, it uses an auxiliary loss. weights_backbone (:class:`~torchvision.models.ResNet50_Weights`, optional): The pretrained weights for the backbone. **kwargs: parameters passed to the ``torchvision.models.segmentation.fcn.FCN`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.segmentation.FCN_ResNet50_Weights :members: NrUr:rdTFTTrareplace_stride_with_dilationrc check_hash) rverifyrrlenrNrr^load_state_dictget_state_dictrarcrUrdrbrerTmodels r$rrusP#))'22G'./?@@+M;GLYeLfHgHghh (XtDD    0ObObObcccH +x 8 8E g44hSW4XXYYY Lr#c t|}tj|}|=d}td|t |jd}td|d}n|d}t |gd}t|||}|*|| |d |S) aaFully-Convolutional Network model with a ResNet-101 backbone from the `Fully Convolutional Networks for Semantic Segmentation `_ paper. .. betastatus:: segmentation module Args: weights (:class:`~torchvision.models.segmentation.FCN_ResNet101_Weights`, optional): The pretrained weights to use. See :class:`~torchvision.models.segmentation.FCN_ResNet101_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. num_classes (int, optional): number of output classes of the model (including the background). aux_loss (bool, optional): If True, it uses an auxiliary loss. weights_backbone (:class:`~torchvision.models.ResNet101_Weights`, optional): The pretrained weights for the backbone. **kwargs: parameters passed to the ``torchvision.models.segmentation.fcn.FCN`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.segmentation.FCN_ResNet101_Weights :members: NrUr:rdTrgrhrirk) rrmrrrnrNrr^rorprqs r$rrsP$**733G(/0@AA+M;GLYeLfHgHghh (XtDD   !1PcPcPcdddH +x 8 8E g44hSW4XXYYY Lr#)( functoolsrtypingrrtorchrtransforms._presetsr_apir r r _metar _utilsrrrresnetrrrrrr__all__r Sequentialr&rPrrr8boolr^rQ IMAGENET1K_V1rrr"r#r$rs 7777777777777777######\\\\\\\\\\UUUUUUUUUUUUUU,,,,,, d c c     "   & " " " " "bm " " "" &&&&&;&&&*&&&&&K&&&* 5 5 5 $ 5 5 5 5 5 /G H+-=-KL /3!%#3C3Q 333 * +33# 3 tn 3 /0 33 333  3l 0H I+->-LM 04!%#4E4S 333 + ,33# 3 tn 3 01 33 333  333r#