def test_collapse_video_pyramid(self):
fps = utils.get_fps(test_video_input_path)
laplacian_pyramid = pyramids.laplacian_video_pyramid(
test_video_input_path)
collapsed_pyramid = pyramids.collapse_video_pyramid(laplacian_pyramid)
utils.write_video(collapsed_pyramid,
test_output_path + 'baby-collapse-video-pyramid.mp4',
fps)
def test_write_video(self):
video = utils.read_video(test_video_input_path)
fps = utils.get_fps(test_video_input_path)
utils.write_video(video, test_output_path + 'baby-write-test.mp4', fps)
video_test = utils.read_video(test_output_path + 'baby-write-test.mp4')
length, height, width, pixel = video_test.shape
self.assertTrue(width == 960 and height == 544 and length == 301
and pixel == 3)
def test_filter_video_pyramid(self):
fps = utils.get_fps(test_video_input_path)
laplacian_pyramid = pyramids.laplacian_video_pyramid(
test_video_input_path)
filtered_pyramid = filters.filter_video_pyramid(
laplacian_pyramid[1:-1], fps, 0.4, 3, 0, 10)
collapsed_pyramid = pyramids.collapse_video_pyramid(filtered_pyramid)
utils.write_video(
collapsed_pyramid,
test_output_path + 'baby-collapse-filtered-video-pyramid.mp4', fps)
def test_get_fps(self):
self.assertTrue(utils.get_fps(test_video_input_path) == 30.0)
def test_temporal_bandpass_filter(self):
pyramid = pyramids.laplacian_video_pyramid(test_video_input_path)
fps = int(utils.get_fps(test_video_input_path))
filtered_video = filters.temporal_bandpass_filter(pyramid[3], fps)
utils.write_video(filtered_video,
test_output_path + 'baby-time-filtered.mp4', fps)
def test_laplacian_video_pyramid(self):
fps = utils.get_fps(test_video_input_path)
laplacians = pyramids.laplacian_video_pyramid(test_video_input_path)
utils.write_video(laplacians[0],
test_output_path + 'baby-laplacian.mp4', fps)