
    h                     n    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y	)
    )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                 b    t         |           t        |        || _        || _        || _        y N)super__init__r	   r   r   r   )selfr   r   r   	__class__s       d/var/www/html/eduruby.in/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   dictstrr'   __classcell__)r   s   @r   r   r   
   s^    %&M- -		 -S[\^\e\eSf -rv - Df$5 r   r   N)collectionsr   typingr   torchr   r   torch.nnr   r"   utilsr	   r,   r    r   r   <module>r6      s$    #   $ (ryy r   