本文借鉴了https://blog.csdn.net/qq_42489503/article/details/107593229。
并且在此之上做了相关的修改优化,在这说下我的配置 tensorflow——gpu版本1.13.1。相关的问题,我都在下面的代码中做了详解。
# 开发者: lin
# 开发时间:2021/7/28 10:41
import imageio
import time
import os
import PIL
import sys
import scipy.io
import matplotlib.pyplot as plt
from matplotlib.pyplot import imshow
import nst_utils
import numpy as np
import tensorflow as tf
'''
解决报错问题:ValueError: Shape must be rank 1 but is rank 0 for ‘bn_conv1/Reshape_4’ (op: ‘Reshape’) with input shapes: [1,1,1,64], [].
参考网址:https://blog.csdn.net/qq_34890702/article/details/105917102
'''
#消除warnning
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
#定义内容代价函数
def compute_content_cost(a_C,a_G):
'''
计算内容代价函数
:param a_C: 维度为(1,n_H,n_W,n_C),表示图像c中的隐藏层激活值
:param a_G: 维度为(1,n_H,n_W,n_C&