
    QЦi                     r    S SK Jr  S SKJrJr  S SKJrJr  S SKJ	r
  SSKJr   " S S\R                  5      rg	)
    )OrderedDict)DictOptional)nnTensor)
functional   )_log_api_usage_oncec            	          ^  \ rS rSrS/rSS\R                  S\R                  S\\R                     SS4U 4S jjjrS\	S\
\\	4   4S	 jrS
rU =r$ )_SimpleSegmentationModel
   aux_classifierNbackbone
classifierreturnc                 \   > [         TU ]  5         [        U 5        Xl        X l        X0l        g N)super__init__r
   r   r   r   )selfr   r   r   	__class__s       e/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/torchvision/models/segmentation/_utils.pyr   !_SimpleSegmentationModel.__init__   s&    D! $,    xc                 4   UR                   SS  nU R                  U5      n[        5       nUS   nU R                  U5      n[        R
                  " XSSS9nXS'   U R                  b0  US   nU R                  U5      n[        R
                  " XSSS9nXS'   U$ )NoutbilinearF)sizemodealign_cornersaux)shaper   r   r   Finterpolater   )r   r   input_shapefeaturesresults        r   forward _SimpleSegmentationModel.forward   s    ggbcl==#UOOOAMM!JeTu*A##A&Aa
RWXA5Mr   )r   r   r   r   )__name__
__module____qualname____firstlineno____constants__r   Moduler   r   r   r   strr*   __static_attributes____classcell__)r   s   @r   r   r   
   sh    %&M- -		 -S[\^\e\eSf -rv - - Df$5  r   r   N)collectionsr   typingr   r   torchr   r   torch.nnr   r%   utilsr
   r1   r    r   r   <module>r;      s$    # !  $ (ryy r   