cd8217b051fab48c4155dc20ad58d6cf7f6fc3a2,evaluate.py,,main,#Any#,19

Before Change

	model.build_summary_saver()
	model.init_all_variables(load_model_name=FLAGS.load_model_name)

	total_psnr = total_mse = 0

	for filename in test_filenames:
		mse = model.do_super_resolution_for_evaluate(filename, output_folder=FLAGS.output_dir, output=FLAGS.save_results)
		total_mse += mse
		total_psnr += util.get_psnr(mse)

	logging.info("\n=== Average [%s] MSE:%f, PSNR:%f ===" % (
		FLAGS.test_dataset, total_mse / len(test_filenames), total_psnr / len(test_filenames)))


if __name__ == "__main__":

After Change

	for i in range(FLAGS.tests):
		model.load_model(FLAGS.load_model_name, i, True if FLAGS.tests > 1 else False)
		for test_data in test_list:
			test(model, test_data)


def test(model, test_data):

Instances