U
    jg                     @   s8   d dl Z d dlZd dlZejejfddZdddZdS )    Nc              
   C   sT   zt | |}t||}|W S  tk
rN } zt| W Y d S d }~X Y nX d S )N)npfromfilecv2Zimdecode	Exceptionprint)filenameflagsdtypenimge r   '/home/ubuntu/vocal-remover/lib/utils.pyimread   s    r   c              
   C   s   zXt j| d }t|||\}}|rPt| dd}|| W 5 Q R X W dS W dS W n0 tk
r } zt| W Y dS d }~X Y nX d S )N   zw+b)modeTF)	ospathsplitextr   Zimencodeopentofiler   r   )r   r   paramsextresultr
   fr   r   r   r   imwrite   s    
r   )N)r   r   numpyr   ZIMREAD_COLORuint8r   r   r   r   r   r   <module>   s   
