From 7b869a3a01001754fa0ef76a8b0a69326871267a Mon Sep 17 00:00:00 2001 From: Nick Cammarata Date: Sat, 3 Aug 2019 10:56:54 -0700 Subject: [PATCH] Incorrect variable reference in param.image alpha --- lucid/optvis/param/images.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lucid/optvis/param/images.py b/lucid/optvis/param/images.py index 09435c90..cf85e08e 100644 --- a/lucid/optvis/param/images.py +++ b/lucid/optvis/param/images.py @@ -35,10 +35,10 @@ def image(w, h=None, batch=None, sd=None, decorrelate=True, fft=True, alpha=Fals output = to_valid_rgb(t[..., :3], decorrelate=decorrelate, sigmoid=True) if alpha: a = tf.nn.sigmoid(t[..., 3:]) - return tf.concat([rgb, a], -1) + return tf.concat([output, a], -1) return output def grayscale_image_rgb(*args, **kwargs): """Takes same arguments as image""" output = image(*args, channels=1, **kwargs) - return tf.tile(output, (1,1,1,3)) \ No newline at end of file + return tf.tile(output, (1,1,1,3))