CT-GAN相关实验,想要利用fid指标进行评价,然后在改写代码的时候出现了一些错误,记录一下解决过程,方便大家解决相同的问题,然后也避免自己遇到相同的问题的时候忘记。
源代码参考
方法1
最简单粗暴的方法就是把batch size 改成1,不过这样计算速度会很慢
方法2
使用版本:python 3.6.5 tensorflow-gpu 1.12.0
对应代码:
def _get_inception_layer(sess):
"""Prepares inception net for batched usage and returns pool_3 layer. """
layername = 'FID_Inception_Net/pool_3:0'
pool3 = sess.graph.get_tensor_by_name(layername)
ops = pool3.graph.get_operations()
for op_idx, op in enumerate(ops):
for o in op.outputs:
shape = o.get_shape()
if shape._dims is not None:
shape = [s.value for s in shape]
new_shape = []