diff options
| author | jules@lens <julescarbon@gmail.com> | 2018-05-05 19:32:25 +0200 |
|---|---|---|
| committer | jules@lens <julescarbon@gmail.com> | 2018-05-05 19:32:25 +0200 |
| commit | 93df56d88da5e2390e0d542ed37a57d48f3f3105 (patch) | |
| tree | 9144319dbe9fb2f70f03fea0c68cf043db38c585 /Codes/utils.py | |
| parent | 91e5f1ffb152e1b729fe9d530d9f01e73017abbf (diff) | |
Diffstat (limited to 'Codes/utils.py')
| -rw-r--r-- | Codes/utils.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/Codes/utils.py b/Codes/utils.py index 8466d08..7f54fe7 100644 --- a/Codes/utils.py +++ b/Codes/utils.py @@ -76,7 +76,7 @@ class DataLoader(object): video_name = video.split('/')[-1] self.videos[video_name] = {} self.videos[video_name]['path'] = video - self.videos[video_name]['frame'] = glob.glob(os.path.join(video, '*.jpg')) + self.videos[video_name]['frame'] = glob.glob(os.path.join(video, '*.png')) self.videos[video_name]['frame'].sort() self.videos[video_name]['length'] = len(self.videos[video_name]['frame']) |