o
    ߥi                     @   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__ k/home/ubuntu/.local/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	   torchno_grad)	r
   framesframe1frame2frame3F1_outF2_outF3_outoutr   r   r   forward   s   







		zDeinterlaceNet.forward)__name__
__module____qualname__r   r   __classcell__r   r   r   r   r      s    r   )torch.nnnn1modelscope.models.cv.video_deinterlace.models.enhr   1modelscope.models.cv.video_deinterlace.models.frer   Moduler   r   r   r   r   <module>   s   