
    %`i                     l    d dl mZ d dl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)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       z/home/jaya/work/projects/VOICE-AGENT/VIET/agent-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   dictstrr'   __classcell__)r   s   @r   r   r   
   s        %&M- - -	 -S[\^\eSf -rv - - - - - - Df$5        r   r   N)collectionsr   typingr   torchr   r   torch.nnr   r"   utilsr	   r,   r    r   r   <module>r6      s    # # # # # #               $ $ $ $ $ $ ( ( ( ( ( (    ry     r   