def test_modify_placement(self):
fLOG(
__file__,
self._testMethodName,
OutputPrint=__name__ == "__main__")
temp = get_temp_folder(__file__, "temp_video_placement")
vid = video_load(os.path.join(temp, '..', 'data', 'videxa.mp4'))
vid2 = video_compose(vid, vid, t2=1, place='h2')
exp = os.path.join(temp, "h2.mp4")
video_save(vid2, exp)
self.assertExists(exp)
clean_video([vid2, vid])
vid = video_load(os.path.join(temp, '..', 'data', 'videxa.mp4'))
vid2 = video_compose(vid, vid, t2=1, place='v2')
exp = os.path.join(temp, "v2.mp4")
video_save(vid2, exp)
self.assertExists(exp)
clean_video([vid2, vid])
vid = video_load(os.path.join(temp, '..', 'data', 'videxa.mp4'))
vid2 = video_compose(vid, vid, t2=1, place='br', zoom=0.2)
exp = os.path.join(temp, "br.mp4")
video_save(vid2, exp)
self.assertExists(exp)
clean_video([vid2, vid])
def test_extract_frames_check_open(self):
fLOG(__file__,
self._testMethodName,
OutputPrint=__name__ == "__main__")
vid = video_load(
os.path.join(os.path.dirname(__file__), 'data', 'videxa.mp4'))
fra = list(video_enumerate_frames(vid, clean=True))
self.assertEqual(len(fra), 78)
self.assertEqual(fra[0].shape, (720, 404, 3))
clean_video(vid)
def test_modify_avideo(self):
fLOG(__file__,
self._testMethodName,
OutputPrint=__name__ == "__main__")
temp = get_temp_folder(__file__, "temp_video_modification")
vid = video_load(os.path.join(temp, '..', 'data', 'videxa.mp4'))
vid2 = video_modification(vid, speed=2., mirrory=True, mirrorx=True)
exp = os.path.join(temp, "courte2x.mp4")
video_save(vid2, exp)
self.assertExists(exp)
clean_video([vid2, vid])
def test_modify_avideo_blur(self):
temp = get_temp_folder(__file__, "temp_videodl_face_blur")
vid = video_load(os.path.join(temp, '..', 'data', 'charlie.mp4'))
vide = video_extract_video(vid, 0, 5 if __name__ == "__main__" else 1)
vid2 = video_map_images(
vide,
fps=10,
name="detect",
logger='bar' if __name__ == "__main__" else None)
exp = os.path.join(temp, "face.mp4")
video_save(vid2, exp)
self.assertExists(exp)
clean_video([vid2, vid, vide])
def test_modify_avideo(self):
temp = get_temp_folder(__file__, "temp_videodl_face_rect")
vid = video_load(os.path.join(temp, '..', 'data', 'charlie.mp4'))
vide = video_extract_video(vid, 0, 10 if __name__ == "__main__" else 1)
vid2 = video_map_images(
vide,
fps=10,
name="detect",
action="rect",
logger='bar' if __name__ == "__main__" else None)
exp = os.path.join(temp, "face.mp4")
self.assertTrue(vid2.make_frame is not None)
video_save_image(vid2, t=1, filename=os.path.join(temp, "img1.jpg"))
im = video_save_image(vid2, t=2)
im.save(os.path.join(temp, "img2.png"))
video_save(vid2, exp, verbose=__name__ == "__main__")
self.assertExists(exp)
clean_video([vid2, vid, vide])