
    jC                     P    d dl Z d dlmZ d dlZddlmZ dgZ G d de          ZdS )    N)Any   )AbstractDatasetOrientationDatasetc                   0     e Zd ZdZdededdf fdZ xZS )r   an  Implements a basic image dataset where targets are filled with zeros.

    >>> from doctr.datasets import OrientationDataset
    >>> train_set = OrientationDataset(img_folder="/path/to/images")
    >>> img, target = train_set[0]

    Args:
        img_folder: folder with all the images of the dataset
        **kwargs: keyword arguments from `AbstractDataset`.
    
img_folderkwargsreturnNc                      t                      j        |fi | d t          j        | j                  D             | _        d S )Nc                 <    g | ]}|t          j        d g          fS )r   )nparray).0img_names     ]/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/doctr/datasets/orientation.py
<listcomp>z/OrientationDataset.__init__.<locals>.<listcomp>'   s(    2s2s2sQYHbhsmm3L2s2s2s    )super__init__oslistdirrootdata)selfr   r	   	__class__s      r   r   zOrientationDataset.__init__   s\    
 		
 	
	
 	
 	
 3t2s]_]ghlhq]r]r2s2s2s			r   )__name__
__module____qualname____doc__strr   r   __classcell__)r   s   @r   r   r      sp        	 	tt t 
	t t t t t t t t t tr   )	r   typingr   numpyr   datasetsr   __all__r    r   r   <module>r'      s    
			           % % % % % %
 t t t t t t t t t tr   