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
