makedirs

1 files changed, 3 insertions, 4 deletions

diff --git a/test.py b/test.py
index d89f2e2..aaa2ff5 100644
--- a/test.py
+++ b/test.py
@@ -41,8 +41,8 @@ def process_dir(v, scene_name=None):
         scene_name = v.split('/')[-1]
     os.makedirs('./results/{}L/L/{}/'.format(args.L, scene_name))
     
-    dir_frames = glob.glob(v + '/*.png')
-    dir_frames.sort()
+    dir_frames = sorted(glob.glob(v + '/*.png'))
+    print(dir_frames)
 
     frames = []
     for f in dir_frames:
@@ -157,6 +157,5 @@ with tf.Session(config=config) as sess:
         if args.dir:
             process_dir(args.dir, scene_name=args.dir.replace('.', '').replace('/', '_'))
         else:
-            dir_inputs = glob.glob('./inputs/L/*')
-            for v in dir_inputs:
+            for v in sorted(glob.glob('./inputs/L/*')):
                 process_dir(v)