wahaha commited on
Commit
149e6d6
1 Parent(s): 8357abe
Files changed (1) hide show
  1. test1.py +2 -0
test1.py CHANGED
@@ -55,10 +55,12 @@ def stats_graph(graph):
55
 
56
  g_sess = None
57
  test_generated = None
 
58
 
59
  def test(checkpoint_dir, style_name, test_file, if_adjust_brightness, img_size=[256,256]):
60
  global g_sess
61
  global test_generated
 
62
 
63
  # tf.reset_default_graph()
64
  result_dir = 'results/'+style_name
 
55
 
56
  g_sess = None
57
  test_generated = None
58
+ test_real = None
59
 
60
  def test(checkpoint_dir, style_name, test_file, if_adjust_brightness, img_size=[256,256]):
61
  global g_sess
62
  global test_generated
63
+ global test_real
64
 
65
  # tf.reset_default_graph()
66
  result_dir = 'results/'+style_name