sess.close()
if compute_running_time:
##############################
# 3 Computing running time #
##############################
print("Evaluating model speed")
print("This can take a few minutes\n")
tf.reset_default_graph()
print("Testing pre-trained baseline SRCNN model")
avg_time_baseline, max_ram = utils.compute_running_time(
"superres", "models_pretrained/div2k_srcnn.pb",
test_data_dir + "HD_res/")
tf.reset_default_graph()
print("Testing provided model")
avg_time_solution, max_ram = utils.compute_running_time(
"superres", "models_converted/model.pb", test_data_dir + "HD_res/")
print("Baseline SRCNN time, ms: ", avg_time_baseline)
print("Test model time, ms: ", avg_time_solution)
print("Speedup ratio (baseline, ms / solution, ms): %.4f" %
(float(avg_time_baseline) / avg_time_solution))
print("Approximate RAM consumption (HD image): " + str(max_ram) +
" MB")
print("\n-------------------------------------\n")
sess.close()
if compute_running_time:
##############################
# 3 Computing running time #
##############################
print("Evaluating model speed")
print("This can take a few minutes\n")
tf.reset_default_graph()
print("Testing pre-trained baseline SRCNN model")
avg_time_baseline, max_ram = utils.compute_running_time(
"superres", "ai_challenge/models_pretrained/dped_srcnn.pb",
"ai_challenge/dped/HD_res/")
tf.reset_default_graph()
print("Testing provided model")
avg_time_solution, max_ram = utils.compute_running_time(
"superres", "ai_challenge/models_converted/model.pb",
"ai_challenge/dped/HD_res/")
print("Baseline SRCNN time, ms: ", avg_time_baseline)
print("Test model time, ms: ", avg_time_solution)
print("Speedup ratio (baseline, ms / solution, ms): %.4f" %
(float(avg_time_baseline) / avg_time_solution))
print("Approximate RAM consumption (HD image): " + str(max_ram) +
" MB")
# sess.close()
if compute_running_time:
##############################
# 3 Computing running time #
##############################
print("Evaluating model speed")
print("This can take a few minutes\n")
tf.reset_default_graph()
print("Testing pre-trained baseline SRCNN model")
avg_time_baseline, max_ram = utils.compute_running_time(
"superres", "models_pretrained/dped_srcnn.pb",
"/root/Desktop/valid/HD_res/")
tf.reset_default_graph()
print("Testing provided model")
avg_time_solution, max_ram = utils.compute_running_time(
"superres", "models_converted/model.pb", "dped/HD_res/")
print("Baseline SRCNN time, ms: ", avg_time_baseline)
print("Test model time, ms: ", avg_time_solution)
print("Speedup ratio (baseline, ms / solution, ms): %.4f" %
(float(avg_time_baseline) / avg_time_solution))
print("Approximate RAM consumption (HD image): " + str(max_ram) +
" MB")