diff options
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']) |