尝试在 jyputer 笔记本中运行 styleGAN，它显示“tensorflow”没有属性“Dimension”

Question

在这里输入图像描述！pythonencode_images.py --optimizer=lbfgs --face_mask=True --iterations=6 --use_lpips_loss=0 --use_discriminator_loss=0 --output_video=Truealigned_images/ generated_images/ Latent_representations/

print("\n************ 潜在代码优化完成！****************")

2021-08-24 13:33:11.033451：我tensorflow/stream_executor/platform/default/dso_loader.cc:48]成功打开动态库cudart64_101.dll

回溯（最近一次调用最后一次）：文件“encode_images.py”，第 12 行，导入 dnnlib.tflib as tflib 文件“C:\Users\bkvij\Office Rapid Innovation\StyleGAN Face Morphing - Arxiv Insights\stylegan-encoder\dnnlib \tflib_ init _.py”，第 8 行，在

从 。导入自动摘要文件“C:\Users\bkvij\Office Rapid Innovation\StyleGAN Face Morphing - Arxiv Insights\stylegan-encoder\dnnlib\tflib\autosummary.py”，第 31 行，位于

从 。导入 tfutil 文件“C:\Users\bkvij\Office Rapid Innovation\StyleGAN Face Morphing - Arxiv Insights\stylegan-encoder\dnnlib\tflib\tfutil.py”，第 34 行，位于

def shape_to_list(shape: Iterable[tf.Dimension]) -> List[Union[int, None]]: AttributeError: 模块“tensorflow”没有属性“Dimension”

Answer 1

这是因为 tf.Dimension 已被弃用。

转到第 34 行stylegan/dnnlib/tflib/tfutil.py并将其更改为。tf.Dimensiontf.compat.v1.Dimension