python - 洋红色多样式拼贴生成器仅生成纯黑色图像？

Question

我一直在使用 Magenta Multistyle Pastiche Generator，但到目前为止它只会生成纯黑色图像。GitHub 论坛上有人遇到了同样的问题，显然 Magenta 只在 CPU 上与 TensorFlow 1.0.1 兼容？但是，我安装了 1.0.1，我仍然遇到同样的问题。这是我在终端中运行的内容（当然，在“源激活洋红色”之后）

$ image_stylization_transform \

     --num_styles=10 \
     --checkpoint=/Users/NMB1331/Desktop/multistyle-pastiche-generator-monet.ckpt \
     --input_image=/Users/NMB1331/Desktop/test_image.jpg \
     --which_styles="{0:0.1,1:0.1,2:0.1,3:0.1,4:0.1,5:0.1,6:0.1,7:0.1,8:0.1,9:0.1}" \
     --output_dir=/Users/NMB1331/Desktop/PracticeOutput \
     --output_basename="all_monet_styles2"

任何人都可以帮助我，或者将我引导到可以解决此问题的地方吗？我也尝试将 magenta 和 tensorflow 升级到最新版本，但无济于事。谢谢！

Answer 1

我改变了环境，docker而不是pip install magenta，解决了这个问题。 https://github.com/tensorflow/magenta-demos/pull/18#issuecomment-334453199

希望这些信息对您有所帮助。