def test(sess, input_z, out_channel_dim, epoch):

print ("input_z =",input_z)

import cv2
from numpy import array
import glob, os
os.chdir("/content/drive/My Drive/Colab Notebooks/test/BlackBox_image")
for name in glob.glob("*.jpg"):
%cd /content/drive/My Drive/Colab Notebooks/test/BlackBox_image
print(name)
im = cv2.imread(name)
maskname='Mask'+name
%cd /content/drive/My Drive/Colab Notebooks/test/Mask/
mask=cv2.imread(maskname)
print(maskname)

mask = cv2.cvtColor(mask, cv2.COLOR_BGR2GRAY)

mask = array(mask).reshape(1, 256,256,1)

image_mask = tf.concat([image, mask], axis=-1)

samples = sess.run(generator(input_z, out_channel_dim, False),
feed_dict={input_z: image})
#print ("batch_images =",batch_images.shape())
sample_images = [((sample + 1.0) * 127.5).astype(np.uint8) for sample in
samples]
show_samples(sample_images, OUTPUT_DIR + "samples", epoch)