
    h                     n    S SK Jr  S SK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)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       ^/var/www/auris/envauris/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   dictstrr)   __static_attributes____classcell__)r   s   @r   r   r   
   sh    %&M- -		 -S[\^\e\eSf -rv - - Df$5  r   r   N)collectionsr   typingr   torchr   r   torch.nnr   r$   utilsr	   r0   r    r   r   <module>r;      s$    #   $ (ryy r   