diff options
Diffstat (limited to 'test/biggan_test.py')
| -rw-r--r-- | test/biggan_test.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/test/biggan_test.py b/test/biggan_test.py index db3e59f..b1a5b85 100644 --- a/test/biggan_test.py +++ b/test/biggan_test.py @@ -2,6 +2,7 @@ import unittest from scipy.stats import truncnorm import numpy as np from gantools import biggan +from gantools import image_utils def create_random_input(dim_z, vocab_size, batch_size=1, truncation = 0.5, rand_seed = 123): def one_hot(index, dim): @@ -20,6 +21,7 @@ class TestBigGAN(unittest.TestCase): gan = biggan.BigGAN() vectors, labels, truncation = create_random_input(gan.dim_z, gan.vocab_size) ims = gan.sample(vectors, labels, truncation) + image_utils.save_images(ims) if __name__ == '__main__': |