• 大小: 1.88MB
    文件类型: .zip
    金币: 1
    下载: 0 次
    发布日期: 2023-10-25
  • 语言: Python
  • 标签: style  transfer  

资源简介

python+tensorflow style transfer 代码,自己添加文件,运行方式,python nueral_style.py --content --styles <styleimg> --output

资源截图

代码片段和文件信息

# Copyright (c) 2015-2018 Anish Athalye. Released under GPLv3.

import os

import numpy as np
import scipy.misc

from stylize import stylize

import math
from argparse import ArgumentParser

from PIL import Image

# default arguments
CONTENT_WEIGHT = 5e0
CONTENT_WEIGHT_BLEND = 1
style_WEIGHT = 5e2
TV_WEIGHT = 1e2
style_layer_WEIGHT_EXP = 1
LEARNING_RATE = 1e1
BETA1 = 0.9
BETA2 = 0.999
EPSILON = 1e-08
style_SCALE = 1.0
ITERATIONS = 1000
VGG_PATH = ‘imagenet-vgg-verydeep-19.mat‘
POOLING = ‘max‘

def build_parser():
    parser = ArgumentParser()
    parser.add_argument(‘--content‘
            dest=‘content‘ help=‘content image‘
            metavar=‘CONTENT‘ required=True)
    parser.add_argument(‘--styles‘
            dest=‘styles‘
            nargs=‘+‘ help=‘one or more style images‘
            metavar=‘style‘ required=True)
    parser.add_argument(‘--output‘
            dest=‘output‘ help=‘output path‘
            metavar=‘OUTPUT‘ required=True)
    parser.add_argument(‘--iterations‘ type=int
            dest=‘iterations‘ help=‘iterations (default %(default)s)‘
            metavar=‘ITERATIONS‘ default=ITERATIONS)
    parser.add_argument(‘--print-iterations‘ type=int
            dest=‘print_iterations‘ help=‘statistics printing frequency‘
            metavar=‘PRINT_ITERATIONS‘)
    parser.add_argument(‘--checkpoint-output‘
            dest=‘checkpoint_output‘ help=‘checkpoint output format e.g. output%%s.jpg‘
            metavar=‘OUTPUT‘)
    parser.add_argument(‘--checkpoint-iterations‘ type=int
            dest=‘checkpoint_iterations‘ help=‘checkpoint frequency‘
            metavar=‘CHECKPOINT_ITERATIONS‘)
    parser.add_argument(‘--width‘ type=int
            dest=‘width‘ help=‘output width‘
            metavar=‘WIDTH‘)
    parser.add_argument(‘--style-scales‘ type=float
            dest=‘style_scales‘
            nargs=‘+‘ help=‘one or more style scales‘
            metavar=‘style_SCALE‘)
    parser.add_argument(‘--network‘
            dest=‘network‘ help=‘path to network parameters (default %(default)s)‘
            metavar=‘VGG_PATH‘ default=VGG_PATH)
    parser.add_argument(‘--content-weight-blend‘ type=float
            dest=‘content_weight_blend‘ help=‘content weight blend conv4_2 * blend + conv5_2 * (1-blend) (default %(default)s)‘
            metavar=‘CONTENT_WEIGHT_BLEND‘ default=CONTENT_WEIGHT_BLEND)
    parser.add_argument(‘--content-weight‘ type=float
            dest=‘content_weight‘ help=‘content weight (default %(default)s)‘
            metavar=‘CONTENT_WEIGHT‘ default=CONTENT_WEIGHT)
    parser.add_argument(‘--style-weight‘ type=float
            dest=‘style_weight‘ help=‘style weight (default %(default)s)‘
            metavar=‘style_WEIGHT‘ default=style_WEIGHT)
    parser.add_argument(‘--style-layer-weight-exp‘ type=float
            dest=‘style_layer_weight_exp‘ help=‘style layer weight exponentional increase - weight(layer) = weight_exp*weight(layer) (defa

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           0  2018-03-29 02:21  neural-style-master\
     文件         196  2018-03-29 02:21  neural-style-master\.editorconfig
     文件          40  2018-03-29 02:21  neural-style-master\.gitignore
     文件       35147  2018-03-29 02:21  neural-style-master\LICENSE.txt
     文件        5901  2018-03-29 02:21  neural-style-master\README.md
     目录           0  2018-03-29 02:21  neural-style-master\examples\
     文件      245099  2018-03-29 02:21  neural-style-master\examples\1-content.jpg
     文件       64615  2018-03-29 02:21  neural-style-master\examples\1-output.jpg
     文件      562577  2018-03-29 02:21  neural-style-master\examples\1-style.jpg
     文件      120188  2018-03-29 02:21  neural-style-master\examples\2-content.jpg
     文件       44983  2018-03-29 02:21  neural-style-master\examples\2-output.jpg
     文件       21516  2018-03-29 02:21  neural-style-master\examples\2-style1.jpg
     文件      107366  2018-03-29 02:21  neural-style-master\examples\2-style2.jpg
     目录           0  2018-03-29 02:21  neural-style-master\examples\tweaks\
     文件      110363  2018-03-29 02:21  neural-style-master\examples\tweaks\cwe01.jpg
     文件      114640  2018-03-29 02:21  neural-style-master\examples\tweaks\cwe10_default.jpg
     文件      104516  2018-03-29 02:21  neural-style-master\examples\tweaks\swe02.jpg
     文件      113476  2018-03-29 02:21  neural-style-master\examples\tweaks\swe14_pavg.jpg
     文件      132475  2018-03-29 02:21  neural-style-master\examples\tweaks\swe14_pmax.jpg
     文件      119562  2018-03-29 02:21  neural-style-master\examples\tweaks\swe14_pmax_pcyuv.jpg
     文件      111108  2018-03-29 02:21  neural-style-master\examples\tweaks\swe20.jpg
     文件        9016  2018-03-29 02:21  neural-style-master\neural_style.py
     文件         133  2018-03-29 02:21  neural-style-master\requirements.txt
     文件        9949  2018-03-29 02:21  neural-style-master\stylize.py
     文件        2385  2018-03-29 02:21  neural-style-master\vgg.py

评论

共有 条评论