3
`a                 @   sN   d dl mZ d dlZd dlZd dljZd dljZ	d dl
Z
dddZdd	 ZdS )    )ImageDataGeneratorN   TFc             c   s   xt t| D ]}tj| | }tj|d|d}|d }tj||}|sZtj	||j
d n|}|d d d d d df }tj	|d|j
 }|V  qW d S )NZimageio)pluginas_gray         )r   )r   )rangelenbase64	b64decodeioZimreadtransresizenpreshapeshape)img_datatarget_sizeZflag_multi_classr   iZ
img_stringimg r   #/var/www/deploy/public/hair/data.pysingleTestGenerator   s    r   c             C   s   | d d d d d df }|S )Nr   r   )Znpyfiler   r   r   r   
saveResult   s    r   r   r   )r   TF)Zkeras.preprocessing.imager   numpyr   osZ
skimage.ior   Zskimage.transform	transformr   r   r   r   r   r   r   r   <module>   s   


