Spaces:
Runtime error
Runtime error
init
Browse files
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
|