U
    ¯Ê¹j  ã                   @   s   d dl Z d dlZddd„ZdS )é    NÚ c              	   C   sÖ   t | tƒrtstdƒ‚| S | }tj |¡sPtj ||¡}tj |¡sPtd|  ƒ‚tj | ¡d  ¡ }|dkr”t	|ƒ�}dd„ t
|ƒD ƒ}W 5 Q R X n>|dkr¾t	|dƒ�}t |¡}W 5 Q R X nd	sÒtd
|› d�ƒ‚|S )Nz class_map dict must be non-emptyz+Cannot locate specified class map file (%s)éÿÿÿÿz.txtc                 S   s   i | ]\}}|  ¡ |“qS © )Ústrip)Ú.0ÚkÚvr   r   úZ/var/www/html/bwave/python/venv/lib/python3.8/site-packages/timm/data/readers/class_map.pyÚ
<dictcomp>   s      z"load_class_map.<locals>.<dictcomp>z.pklÚrbFz&Unsupported class map file extension (z).)Ú
isinstanceÚdictÚAssertionErrorÚosÚpathÚexistsÚjoinÚsplitextÚlowerÚopenÚ	enumerateÚpickleÚload)Zmap_or_filenameÚrootZclass_map_pathZclass_map_extÚfZclass_to_idxr   r   r	   Úload_class_map   s     

r   )r   )r   r   r   r   r   r   r	   Ú<module>   s   