Skip to content

GitLab

Unverified Commit e5eaf2a7 authored by Jintao Lin's avatar Jintao Lin Committed by GitHub
Browse files

update video/io.py (#856)

parent 48d99025
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 9 deletions
mmcv/video/io.py
View file @ e5eaf2a7
...@@ -225,6 +225,8 @@ class VideoReader: ...@@ -225,6 +225,8 @@ class VideoReader:
def write_frame(file_idx): def write_frame(file_idx):
img = self.read() img = self.read()
if img is None:
return
filename = osp.join(frame_dir, filename_tmpl.format(file_idx)) filename = osp.join(frame_dir, filename_tmpl.format(file_idx))
cv2.imwrite(filename, img) cv2.imwrite(filename, img)
...@@ -233,12 +235,7 @@ class VideoReader: ...@@ -233,12 +235,7 @@ class VideoReader:
file_start + task_num)) file_start + task_num))
else: else:
for i in range(task_num): for i in range(task_num):
img = self.read() write_frame(file_start + i)
if img is None:
break
filename = osp.join(frame_dir,
filename_tmpl.format(i + file_start))
cv2.imwrite(filename, img)
def __len__(self): def __len__(self):
return self.frame_cnt return self.frame_cnt
...@@ -317,7 +314,5 @@ def frames2video(frame_dir, ...@@ -317,7 +314,5 @@ def frames2video(frame_dir,
track_progress(write_frame, range(start, end)) track_progress(write_frame, range(start, end))
else: else:
for i in range(start, end): for i in range(start, end):
filename = osp.join(frame_dir, filename_tmpl.format(i)) write_frame(i)
img = cv2.imread(filename)
vwriter.write(img)
vwriter.release() vwriter.release()
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment