From 5255a009546ed93415e47bdca6a5c36f7ff739e4 Mon Sep 17 00:00:00 2001 From: Matt Cooper Date: Sun, 21 Aug 2016 16:03:14 -0400 Subject: updated test --- Code/test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Code/test.py b/Code/test.py index ec53b66..134b8b9 100644 --- a/Code/test.py +++ b/Code/test.py @@ -36,7 +36,7 @@ def get_full_clips(data_dir, num_clips, num_rec_out=1): # get a random clip of length HIST_LEN + num_rec_out from each episode for clip_num, ep_dir in enumerate(ep_dirs): - ep_frame_paths = glob(os.path.join(ep_dir, '*')) + ep_frame_paths = sorted(glob(os.path.join(ep_dir, '*'))) start_index = np.random.choice(len(ep_frame_paths) - (c.HIST_LEN + num_rec_out - 1)) clip_frame_paths = ep_frame_paths[start_index:start_index + (c.HIST_LEN + num_rec_out)] print clip_num -- cgit v1.2.3-70-g09d2