3
`a                 @   sj   d dl Zd dlZd dljZd dljZd dl Zd dlZd dl	T d dl
T d dlT d dlmZmZ d	ddZdS )
    N)*)ModelCheckpointLearningRateScheduler      c             C   s  t |}tdddddd|}tdddddd|}tdd|}td	ddddd|}td	ddddd|}tdd|}td
ddddd|}td
ddddd|}tdd|}tdddddd|}	tdddddd|	}	td|	}
tdd|
}tdddddd|}tdddddd|}td|}tddddddtdd|}t|
|gdd}tdddddd|}tdddddd|}td
dddddtdd|}t||gdd}td
ddddd|}td
ddddd|}td	dddddtdd|}t||gdd}td	ddddd|}td	ddddd|}tddddddtdd|}t||gdd}tdddddd|}tdddddd|}tdddddd|}tdddd|}t||d}|jdddgd | r|j|  |S ) N@   r   relusame	he_normal)
activationpaddingkernel_initializer   )	pool_size   r   i   g      ?i   )size)axis   sigmoid)r   )inputsoutputsadambinary_crossentropyaccuracy)	optimizerlossmetrics)r   r   )r   r   )r   r   )r   r   )r   r   )r   r   )r   r   )r   r   )	InputConv2DMaxPooling2DDropoutUpSampling2DconcatenateModelcompileload_weights)pretrained_weights
input_sizer   Zconv1Zpool1Zconv2Zpool2Zconv3Zpool3Zconv4Zdrop4Zpool4Zconv5Zdrop5Zup6Zmerge6Zconv6Zup7Zmerge7Zconv7Zup8Zmerge8Zconv8Zup9Zmerge9Zconv9Zconv10model r)   $/var/www/deploy/public/hair/model.pyunet   sP        
r+   r   r   r   )Nr,   )numpynpos
skimage.ioioskimage.transform	transformtranskeraskeras.modelskeras.layersZkeras.optimizersZkeras.callbacksr   r   r+   r)   r)   r)   r*   <module>   s   

