
    kh                     r    d dl mZ d dlmZmZ d dlmZmZ d dlm	Z
 ddlmZ  G d dej                        Zy	)
    )OrderedDict)DictOptional)nnTensor)
functional   )_log_api_usage_oncec            	            e Zd ZdgZd	dej
                  dej
                  deej
                     ddf fdZdede	e
ef   fdZ xZS )
_SimpleSegmentationModelaux_classifierNbackbone
classifierreturnc                 b    t         |           t        |        || _        || _        || _        y N)super__init__r
   r   r   r   )selfr   r   r   	__class__s       b/var/www/teggl/fontify/venv/lib/python3.12/site-packages/torchvision/models/segmentation/_utils.pyr   z!_SimpleSegmentationModel.__init__   s,    D! $,    xc                 B   |j                   dd  }| j                  |      }t               }|d   }| j                  |      }t	        j
                  ||dd      }||d<   | j                  4|d   }| j                  |      }t	        j
                  ||dd      }||d<   |S )NoutbilinearF)sizemodealign_cornersaux)shaper   r   r   Finterpolater   )r   r   input_shapefeaturesresults        r   forwardz _SimpleSegmentationModel.forward   s    ggbcl==#UOOOAMM!+JeTu*A##A&Aak
RWXAF5Mr   r   )__name__
__module____qualname____constants__r   Moduler   r   r   r   strr(   __classcell__)r   s   @r   r   r   
   s^    %&M- -		 -S[\^\e\eSf -rv - Df$5 r   r   N)collectionsr   typingr   r   torchr   r   torch.nnr   r#   utilsr
   r-   r    r   r   <module>r6      s$    # !  $ (ryy r   