diff options
| author | Matt Cooper <matthew_cooper@brown.edu> | 2016-08-21 16:03:14 -0400 |
|---|---|---|
| committer | Matt Cooper <matthew_cooper@brown.edu> | 2016-08-21 16:03:14 -0400 |
| commit | 5255a009546ed93415e47bdca6a5c36f7ff739e4 (patch) | |
| tree | 23f43554565a96f52cd6ed53734defcdf729a774 /Code/test.py | |
| parent | 4315a63c5d0c368075b8f4e1dab0a0711ae4440c (diff) | |
updated test
Diffstat (limited to 'Code/test.py')
| -rw-r--r-- | Code/test.py | 2 |
1 files changed, 1 insertions, 1 deletions
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 |