Skip to content

Commit

Permalink
small fix for ori_writer and slomo_writer
Browse files Browse the repository at this point in the history
  • Loading branch information
duguyue100 committed Jun 17, 2021
1 parent 2c2b995 commit 57af23f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions v2ecore/slomo.py
Original file line number Diff line number Diff line change
Expand Up @@ -466,7 +466,7 @@ def interpolate(self, source_frame_path, output_folder, frame_size):
# write input frames into video
# don't duplicate each frame if called using rotating buffer
# of two frames in a row
if self.ori_writer is None:
if self.ori_writer:
src_files = sorted(
glob.glob("{}".format(source_frame_path) + "/*.npy"))

Expand All @@ -480,7 +480,7 @@ def interpolate(self, source_frame_path, output_folder, frame_size):
self.numOrigVideoFramesWritten += 1

frame_paths = self.__all_images(output_folder)
if self.slomo_writer is None:
if self.slomo_writer:
for path in tqdm(frame_paths,desc='write-slomo-vid',unit='fr'):
frame = self.__read_image(path)
self.slomo_writer.write(
Expand Down

0 comments on commit 57af23f

Please sign in to comment.