From 93df56d88da5e2390e0d542ed37a57d48f3f3105 Mon Sep 17 00:00:00 2001 From: "jules@lens" Date: Sat, 5 May 2018 19:32:25 +0200 Subject: new scripts --- Codes/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'Codes/utils.py') 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']) -- cgit v1.2.3-70-g09d2