if start is not None:
frame_nums_to_write.append((start, end))
num_selected_frames += (end - start)
frames.append(end - start)
if num_selected_frames >= min_frames:
break
frame_nums_to_write.sort(key=lambda x: x[0])
return frame_nums_to_write
if __name__ == "__main__":
folder = 'test_data_3'
video_name = 'test_video_3'
frames_rgb_folder = 'input/test_dataset/{folder}/frames_rgb_test/{video_name}/'.format(
folder=folder, video_name=video_name)
frames_jpg_folder = 'input/test_dataset/{folder}/frames_test/{video_name}/'.format(
folder=folder, video_name=video_name)
audio_file = 'input/test_dataset/{folder}/{video_name}.wav'.format(
folder=folder, video_name=video_name)
evaluator = Evaluator(frames_rgb_folder, audio_file, None)
evaluator.evaluate()
frame_nums_to_write = evaluator.select_frames()
# offline run
converter = VideoConverter(frame_nums_to_write, frames_jpg_folder,
evaluator.audio.data, 30, evaluator.audio.rate,
evaluator.audio.sampwidth)
converter.offline_conversion(
'./output_offline/{video_name}'.format(video_name=video_name))
