
    ڧg                     p    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d	S )
    )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                     t                                                       t          |            || _        || _        || _        d S N)super__init__r
   r   r   r   )selfr   r   r   	__class__s       b/var/www/html/ai-engine/env/lib/python3.11/site-packages/torchvision/models/segmentation/_utils.pyr   z!_SimpleSegmentationModel.__init__   sC    D!!! $,    xc                 `   |j         dd          }|                     |          }t                      }|d         }|                     |          }t	          j        ||dd          }||d<   | j        :|d         }|                     |          }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    gbccl==##UOOOAM!+JeTTTu*A##A&&Aak
RWXXXAF5Mr   r   )__name__
__module____qualname____constants__r   Moduler   r   r   r   strr(   __classcell__)r   s   @r   r   r   
   s        %&M- - -	 -S[\^\eSf -rv - - - - - - Df$5        r   r   N)collectionsr   typingr   r   torchr   r   torch.nnr   r#   utilsr
   r-   r    r   r   <module>r6      s    # # # # # # ! ! ! ! ! ! ! !         $ $ $ $ $ $ ( ( ( ( ( (    ry     r   