Download Fixed Edsr-x3.pb -

[1] Lim, B., et al. "Enhanced deep residual networks for single image super-resolution." CVPRW 2017. [2] TensorFlow Model Export Guide – SavedModel to .pb.

graph = load_pb('EDSR_x3.pb') input_tensor = graph.get_tensor_by_name('input:0') output_tensor = graph.get_tensor_by_name('output:0') lr = cv2.imread('lowres.png') # shape (H, W, 3) lr = cv2.cvtColor(lr, cv2.COLOR_BGR2RGB) lr_input = np.expand_dims(lr, 0) # (1, H, W, 3) Run inference with tf.compat.v1.Session(graph=graph) as sess: sr = sess.run(output_tensor, feed_dict={input_tensor: lr_input}) sr = np.squeeze(sr, 0) # (H 3, W 3, 3) Download Fixed Edsr-x3.pb

import tensorflow as tf import cv2 import numpy as np def load_pb(model_path): with tf.io.gfile.GFile(model_path, 'rb') as f: graph_def = tf.compat.v1.GraphDef() graph_def.ParseFromString(f.read()) with tf.Graph().as_default() as graph: tf.import_graph_def(graph_def, name='') return graph [1] Lim, B

cv2.imwrite('superres.png', cv2.cvtColor(sr, cv2.COLOR_RGB2BGR)) 3) lr = cv2.cvtColor(lr