Example #1
0
def read_image(image_path):
    with tf.compat.v1.Session() as sess:
        data_loader = DataLoader(model_config['preprocess']['input_size'][0],
                                 model_config['preprocess']['input_size'][1])
        image, input_plhd = data_loader.build_preprocess(
            style=model_config['preprocess']['preprocess_type'])
        image_val = sess.run(image, feed_dict={input_plhd: image_path})
    return image_val