o
    *j                     @   s   d dl mZ d dlmZmZ d dlZd dlmZ d dl	m
Z
 d dlmZ d dlmZ d dlmZmZ ejejejdG d	d
 d
e
ZdS )    N)AnyDict)Models)
TorchModel)MODELS)RDE_VOS)	ModelFileTasks)module_namec                       s8   e Zd Zdef fddZdeeef fddZ  ZS )VideoObjectSegmentation	model_dirc                    s\   t  j|g|R i | t|tj}tj|dd}t | _	| j	j
|dd | j	  d S )Ncpu)Zmap_locationT)strict)super__init__ospjoinr   ZTORCH_MODEL_FILEtorchloadr   modelZload_state_dicteval)selfr   argskwargsZ
model_pathparams	__class__ u/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/modelscope/models/cv/video_object_segmentation/model.pyr      s   z VideoObjectSegmentation.__init__inputsc                 C   s
   |  |S )N)r   )r   r   r   r   r   forward   s   
zVideoObjectSegmentation.forward)	__name__
__module____qualname__strr   r   r   r    __classcell__r   r   r   r   r      s    r   )Zos.pathpathr   typingr   r   r   Zmodelscope.metainfor   Zmodelscope.models.baser   Zmodelscope.models.builderr   Z;modelscope.models.cv.video_object_segmentation.eval_networkr   Zmodelscope.utils.constantr   r	   Zregister_moduleZvideo_object_segmentationr   r   r   r   r   <module>   s   