o
    0j                     @   s8   d dl mZ d dlmZ ejdddG dd dZdS )   )	benchmark)VideoReaderNT)nameZis_read_operationc                       s:   e Zd ZdZd fdd	Zdd	 Zd
d Zdd Z  ZS )	ReadVideozLoad video from the file.opencv      Nc                    s    t    t||||d| _d S )N)backendnum_segseg_lensample_type)super__init__r   _video_reader)selfr	   r
   r   r   	__class__ m/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/common/reader/video_reader.pyr      s   
zReadVideo.__init__c                    s    fdd|D S )applyc                    s   g | ]}  |qS r   )_read).0Zvideor   r   r   
<listcomp>!   s    z&ReadVideo.__call__.<locals>.<listcomp>r   )r   Zvideosr   r   r   __call__   s   zReadVideo.__call__c                 C   s
   |  |S )N)_read_video)r   	file_pathr   r   r   r   #   s   
zReadVideo._readc                 C   s$   t | j|}|d u rtd|S )NzVideo read Error)listr   read	Exception)r   Z
video_pathZblobr   r   r   r   &   s   zReadVideo._read_video)r   r   r   N)	__name__
__module____qualname____doc__r   r   r   r   __classcell__r   r   r   r   r      s    r   )Zutils.benchmarkr   Zutils.ior   Ztimeit_with_optionsr   r   r   r   r   <module>   s   