o
    پi                     @   sd   d dl mZ g dZdaetadd ZdefddZd	d
 Z	dddZ
dd Zdd Zdd ZdS )    )deepcopy)get_img_extensionsis_img_extensionset_img_extensionsadd_img_extensionsdel_img_extensions)z.pngz.jpgz.jpegc                    s(   t   t fdd| D at | ad S )Nc                 3   s&    | ]}| vr  |s|V  qd S N)add.0xdedupe T/home/ubuntu/.local/lib/python3.10/site-packages/timm/data/readers/img_extensions.py	<genexpr>   s   $ z"_set_extensions.<locals>.<genexpr>)settupleIMG_EXTENSIONS_IMG_EXTENSIONS_SET)
extensionsr   r   r   _set_extensions
   s   r   r   c                 C   s$   | ot | tot| dko| dS )N   .)
isinstancestrlen
startswith)r   r   r   r   _valid_extension   s   $r   c                 C   s   | t v S r   )r   extr   r   r   r      s   r   Fc                 C   s   t | rtS tS r   )r   r   r   )as_setr   r   r   r      s   r   c                 C   s.   t | sJ | D ]}t|sJ qt|  d S r   )r   r   r   )r   r   r   r   r   r      s   r   c                 C   sD   t | tttfs| f} | D ]}t|sJ qtt|  }t| d S r   )r   listr   r   r   r   r   )r    r   r   r   r   r   r   %   s   r   c                    s8   t  tttfs f t fddtD }t| d S )Nc                 3   s    | ]	}| vr|V  qd S r   r   r
   r   r   r   r   1   s    z%del_img_extensions.<locals>.<genexpr>)r   r"   r   r   r   r   )r    r   r   r   r   r   .   s   r   N)F)copyr   __all__r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   <module>   s    
	