magenta / magenta-demos

Demonstrations of Magenta Models
Apache License 2.0
1.32k stars 419 forks source link

Getting Black Images as Output for Image Stylization #90

Open rgb91 opened 4 years ago

rgb91 commented 4 years ago

I am getting black images after running the code (from Image Stylization) on my local computer. However, the same thing worked perfectly in Google Colab. I am not sure where is the problem.

I installed packages as in the Google Colab Runtime. Here's a list of necessary packages:

tensorflow                     2.2.0
tensorflow-addons              0.10.0
tensorflow-datasets            3.1.0
tensorflow-estimator           2.2.0
tensorflow-gan                 2.0.0
tensorflow-hub                 0.8.0
tensorflow-metadata            0.22.2
tensorflow-probability         0.10.0
magenta                        2.0.2