def _sample_video_clip_score(clips_list, length, scores_list): clip = np.empty(shape=[time_steps, height, width, 3], dtype=np.float32) sample_idx = RNG.randint(length) for t, filename in enumerate(clips_list[sample_idx]): clip[t, ...] = load_frame(filename, height, width) return clip, np.array(scores_list[sample_idx])
def read_video_clip(self, images_paths): video_clip = [] for filename in images_paths: video_clip.append( load_frame(filename, self.resize_height, self.resize_width)) video_clip = np.stack(video_clip, axis=0) return video_clip
def video_clip_generator(): i = 0 while True: video_clips_paths = videos_clips_list[i] video_clips = np.empty(shape=[time_steps, height, width, 3], dtype=np.float32) for t, filename in enumerate(video_clips_paths): video_clips[t, ...] = load_frame(filename, height, width) i = (i + 1) % num_video_clips yield video_clips
def get_video_clip(self, video, start, end, interval=1): # assert video_name in self._videos_info, 'video {} is not in {}!'.format(video_name, self._videos_info.keys()) # assert (start >= 0) and (start <= end) and (end < self._videos_info[video_name]['length']) video_idx = np.arange(start, end, interval) video_clip = np.empty(shape=[len(video_idx), self.resize_height, self.resize_width, 3], dtype=np.float32) for idx, v_idx in enumerate(video_idx): filename = self.test_videos_info[video]['images'][v_idx] video_clip[idx, ...] = load_frame(filename, self.resize_height, self.resize_width) return video_clip
def get_val_video_clip(self, video, start, end, interval=1): video_idx = np.arange(start, end, interval) video_clip = np.empty( shape=[len(video_idx), self.resize_height, self.resize_width, 3], dtype=np.float32) for idx, v_idx in enumerate(video_idx): filename = self.val_videos_info[video]['images'][v_idx] video_clip[idx, ...] = load_frame(filename, self.resize_height, self.resize_width) return video_clip