summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--test.py7
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)