o
    *j                     @   s:   d dl mZ d dlmZ d dlmZ G dd dejZdS )    N)DeinterlaceEnh)DeinterlaceFrec                       s$   e Zd Z fddZdd Z  ZS )DeinterlaceNetc                    s"   t t|   t | _t | _d S N)superr   __init__r   frenetr   enhnet)self	__class__ x/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/modelscope/models/cv/video_deinterlace/deinterlace_arch.pyr   
   s   zDeinterlaceNet.__init__c           	      C   sz   | j   | j  t % |\}}}|  |}|  |}|  |}| |||g}W d    |S 1 s6w   Y  |S r   )r   evalr	   ZtorchZno_grad)	r
   framesZframe1Zframe2Zframe3ZF1_outZF2_outZF3_outoutr   r   r   forward   s   







		zDeinterlaceNet.forward)__name__
__module____qualname__r   r   __classcell__r   r   r   r   r      s    r   )Ztorch.nnnnZ1modelscope.models.cv.video_deinterlace.models.enhr   Z1modelscope.models.cv.video_deinterlace.models.frer   Moduler   r   r   r   r   <module>   s   