def test_cvt2frames(self):
v = cvb.VideoReader(self.video_path)
frame_dir = tempfile.mkdtemp()
v.cvt2frames(frame_dir)
assert os.path.isdir(frame_dir)
for i in range(self.num_frames):
filename = '{}/{:06d}.jpg'.format(frame_dir, i)
assert os.path.isfile(filename)
os.remove(filename)
v = cvb.VideoReader(self.video_path)
v.cvt2frames(frame_dir, show_progress=False)
assert os.path.isdir(frame_dir)
for i in range(self.num_frames):
filename = '{}/{:06d}.jpg'.format(frame_dir, i)
assert os.path.isfile(filename)
os.remove(filename)
v = cvb.VideoReader(self.video_path)
v.cvt2frames(frame_dir,
file_start=100,
filename_tmpl='{:03d}.JPEG',
start=100,
max_num=20)
assert os.path.isdir(frame_dir)
for i in range(100, 120):
filename = '{}/{:03d}.JPEG'.format(frame_dir, i)
assert os.path.isfile(filename)
os.remove(filename)
os.removedirs(frame_dir)
def test_frames2video(self):
v = cvb.VideoReader(self.video_path)
frame_dir = tempfile.mkdtemp()
out_filename = '.cvbase_test.avi'
v.cvt2frames(frame_dir)
assert os.path.isdir(frame_dir)
for i in range(self.num_frames):
filename = '{}/{:06d}.jpg'.format(frame_dir, i)
assert os.path.isfile(filename)
cvb.frames2video(frame_dir, out_filename)
v = cvb.VideoReader(out_filename)
assert v.fps == 30
assert len(v) == self.num_frames
cvb.frames2video(frame_dir,
out_filename,
fps=25,
start=10,
end=50,
show_progress=False)
v = cvb.VideoReader(out_filename)
assert v.fps == 25
assert len(v) == 40
for i in range(self.num_frames):
filename = '{}/{:06d}.jpg'.format(frame_dir, i)
os.remove(filename)
os.removedirs(frame_dir)
os.remove(out_filename)
def test_position(self):
v = cvb.VideoReader(self.video_path)
assert v.position == 0
for _ in range(10):
v.read()
assert v.position == 10
v.get_frame(100)
assert v.position == 100
def test_cut_concat_video(self):
part1_file = '.tmp1.mp4'
part2_file = '.tmp2.mp4'
cvb.cut_video(self.video_path, part1_file, end=3, vcodec='h264')
cvb.cut_video(self.video_path, part2_file, start=3, vcodec='h264')
v1 = cvb.VideoReader(part1_file)
v2 = cvb.VideoReader(part2_file)
assert len(v1) == 75
assert len(v2) == self.num_frames - 75
out_file = 'tmp.mp4'
cvb.concat_video([part1_file, part2_file], out_file)
v = cvb.VideoReader(out_file)
assert len(v) == self.num_frames
os.remove(part1_file)
os.remove(part2_file)
os.remove(out_file)
def test_load(self):
v = cvb.VideoReader(self.video_path)
assert v.width == 294
assert v.height == 240
assert v.fps == 25
assert v.frame_cnt == self.num_frames
assert len(v) == self.num_frames
assert v.opened
import cv2
assert isinstance(v.vcap, type(cv2.VideoCapture()))
def test_resize_video(self):
out_file = '.tmp.mp4'
cvb.resize_video(self.video_path, out_file, (200, 100), quiet=True)
v = cvb.VideoReader(out_file)
assert v.resolution == (200, 100)
os.remove(out_file)
cvb.resize_video(self.video_path, out_file, ratio=2)
v = cvb.VideoReader(out_file)
assert v.resolution == (294 * 2, 240 * 2)
os.remove(out_file)
cvb.resize_video(self.video_path, out_file, (1000, 480), keep_ar=True)
v = cvb.VideoReader(out_file)
assert v.resolution == (294 * 2, 240 * 2)
os.remove(out_file)
cvb.resize_video(self.video_path,
out_file,
ratio=(2, 1.5),
keep_ar=True)
v = cvb.VideoReader(out_file)
assert v.resolution == (294 * 2, 360)
os.remove(out_file)
def test_read(self):
v = cvb.VideoReader(self.video_path)
img = v.read()
assert int(round(img.mean())) == 94
img = v.get_frame(64)
assert int(round(img.mean())) == 94
img = v[65]
assert int(round(img.mean())) == 205
img = v[64]
assert int(round(img.mean())) == 94
img = v.read()
assert int(round(img.mean())) == 205
with pytest.raises(ValueError):
v.get_frame(self.num_frames + 1)
def test_with(self):
with cvb.VideoReader(self.video_path) as v:
assert v.opened
assert not v.opened
def test_iterator(self):
cnt = 0
for img in cvb.VideoReader(self.video_path):
cnt += 1
assert img.shape == (240, 294, 3)
assert cnt == self.num_frames
def test_current_frame(self):
v = cvb.VideoReader(self.video_path)
assert v.current_frame() is None
v.read()
img = v.current_frame()
assert int(round(img.mean())) == 94
