fast style transfer 快速风格转换 train

最新推荐文章于 2023-02-28 17:16:50 发布

阿豪boy

最新推荐文章于 2023-02-28 17:16:50 发布

阅读量113

点赞数

文章标签： python tensorflow 深度学习机器学习人工智能

本文链接：https://blog.csdn.net/qq_35516360/article/details/122065763

版权

本文介绍了快速风格转换的训练过程，采用单批次训练，每轮仅使用一张图片，并利用Adam优化器和固定学习率进行优化。训练过程大约在1000次迭代时耗时15分钟。

摘要由CSDN通过智能技术生成

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

上一篇: fast style transfer 快速风格转换 model

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

单批次训练

每次只使用一张图片进行训练,在训练中显示中间结果和保存网络模型

使用adam优化器,固定学习率

训练比较耗费时间,1000次大概十五分钟

train

import matplotlib.pyplot as plt
from model import *
import data
from tensorflow.python.framework import graph_util
from config import *
import scipy.misc as sm


def main():
    iterator = data.get_iterator()
    image_batch = iterator.get_next()
    style_images = [
        sm.imread(path, mode='RGB')
        for path in STYLE_IMAGE_PATHS
    ]

    test_images = [
        sm.imresize(sm.imread(path, mo