keras 的 example 文件 neural_doodle.py 解析

这是一个电脑学习涂鸦的程序

在不指定 content-image 的时候，运行指令如下：

python neural_doodle.py --nlabels 4 --style-image Monet/style.png --style-mask Monet/style_mask.png --target-mask Monet/target_mask.png --target-image-prefix generated/monet

在指定 content-image 的时候，运行指令如下：

python neural_doodle.py --nlabels 4 --style-image Renoir/style.png --style-mask Renoir/style_mask.png --target-mask Renoir/target_mask.png --content-image Renoir/creek.jpg --target-image-prefix generated/renoir

首先看下函数 load_mask_labels，该函数会把 mask 根据颜色分成不同的 label 页，比如会把下图 style_mask.png

分为下面四张：

********************************************

 

********************************************

 

********************************************

 

 

同样的 target_mask.png 也是会像上面的样式分开，其中白色部分，内容为True，黑色部分为False；