fast style transfer 快速风格转换 config

原文链接: fast style transfer 快速风格转换 config

上一篇: docker 配置阿里云镜像加速

下一篇: fast style transfer 快速风格转换 data

将项目需要用的资源作为单独的配置文件同一进行管理
包括图像最大尺寸,pb文件保存路径,cpkt文件位置等

配置文件

import numpy as np

# 批次大小,为1时可以不考虑图像大小,大于1时,需要将输入图像转化为相同大小
BATCH_SIZE = 1

# data流 相关配置
SHUFFLE_SIZE = 256
PREFETCH_SIZE = 256

MAX_IMAGE_SIZE = 240
IMAGE_PAD = 5

# vgg 均值数组
MEAN_PIXEL = np.array([123.68, 116.779, 103.939]).reshape((1, 1, 1, 3))

# 训练配置信息
TRAIN_STEP = 20000
SHOW_STEP = 100
SAVE_STEP = 1000
LEARNING_RATE = .01

# 测试图片路径数组
TEST_IMAGE_PATHS = [
    './images/xjtu.jpg',
    './images/xjtu2.jpg',
    './images/lib.jpg',
    './images/lib2.jpg',
    './images/lyf.jpg',
    './images/hml.jpg',
    './images/dog.jpg',
    './images/tree.jpg',
]

# 不同的loss权重设置
# 风格损失为e10,内容为e6,平滑e7
CONTENT_WEIGHT = 1.
STYLE_WEIGHT = .0001
TV_WEIGHT = .001

#  生成器网络前缀
GENERATOR_SCOPE = 'generator'

# vgg19 ckpt
VGG19_CKPT_PATH = r"D:\data\tf_models\vgg19\vgg_19.ckpt"

# 用于训练的图像文件夹
TRAIN_IMAGE_ROOT = r'D:\data\coco\train2014'
# 保存pb文件
PB_PATH = './pb/starry.pb'

# 风格文件路径,可以使用多风格
STYLE_IMAGE_PATHS = [
    './styled_images/starry.jpg',
    # './styled_images/feathers.jpg',
    # './styled_images/mnls.jpg',
    # './styled_images/candy.jpg',
    # './styled_images/tree.jpg',

]
STYLE_WEIGHTS = [1., 1., 1., 1., 1.]
STYLE_LAYERS = [
    'vgg_19/conv1/conv1_1',
    'vgg_19/conv2/conv2_1',
    'vgg_19/conv3/conv3_1',
    'vgg_19/conv4/conv4_1',
    # 'vgg_19/conv5/conv5_1',
]
STYLE_LAYER_WEIGHTS = [1, 1, 1, 1, 1, ]
CONTENT_LAYERS = [
    # 'vgg_19/conv1/conv1_2',
    # 'vgg_19/conv2/conv2_2',
    'vgg_19/conv3/conv3_4',
    'vgg_19/conv4/conv4_4',
    # 'vgg_19/conv5/conv5_2',
]
CONTENT_LAYER_WEIGHTS = [1., 1., 1., 1., 1., 1.]