"src/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "3dcc75cbd4f924070cf9ebbbc1876ee04a0638c4"
Unverified Commit 8e946635 authored by Dhruv Nair's avatar Dhruv Nair Committed by GitHub
Browse files

Update export to video to support new `tensor_to_vid` function in video pipelines (#6715)

update
parent b09b90e2
...@@ -125,7 +125,10 @@ def export_to_video( ...@@ -125,7 +125,10 @@ def export_to_video(
if output_video_path is None: if output_video_path is None:
output_video_path = tempfile.NamedTemporaryFile(suffix=".mp4").name output_video_path = tempfile.NamedTemporaryFile(suffix=".mp4").name
if isinstance(video_frames[0], PIL.Image.Image): if isinstance(video_frames[0], np.ndarray):
video_frames = [(frame * 255).astype(np.uint8) for frame in video_frames]
elif isinstance(video_frames[0], PIL.Image.Image):
video_frames = [np.array(frame) for frame in video_frames] video_frames = [np.array(frame) for frame in video_frames]
fourcc = cv2.VideoWriter_fourcc(*"mp4v") fourcc = cv2.VideoWriter_fourcc(*"mp4v")
......
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