def _get_png(): """Returns a random png image.""" image = fake_data_utils.get_random_picture(HEIGHT, WIDTH) png = tf.image.encode_png(image) with utils.nogpu_session() as sess: res = sess.run(png) return res
def _get_jpeg(height, width): """Returns jpeg picture.""" image = fake_data_utils.get_random_picture(height, width) jpeg = tf.image.encode_jpeg(image) with utils.nogpu_session() as sess: res = sess.run(jpeg) return res