imageio-ffmpeg added, bugfix

pull/5/head
vahid K. nejad 2023-04-12 17:39:24 +04:00
parent 268c97c528
commit 0142e8b258
3 changed files with 8 additions and 11 deletions

View File

@ -1,5 +0,0 @@
import launch
if not launch.is_installed("imageio"):
if not launch.is_installed("imageio.plugins.ffmpeg"):
launch.run_pip("install imageio[ffmpeg]", "requirements for Infinite-Zoom")

View File

@ -1,5 +1,6 @@
import launch import launch
if not launch.is_installed("imageio"): if not launch.is_installed("imageio"):
if not launch.is_installed("imageio.plugins.ffmpeg"): launch.run_pip("install imageio", "requirements 0 for Infinite-Zoom")
launch.run_pip("install imageio[ffmpeg]", "requirements for Infinite-Zoom") if not launch.is_installed("imageio-ffmpeg"):
launch.run_pip("install imageio-ffmpeg", "requirements 1 for Infinite-Zoom")

View File

@ -3,7 +3,7 @@ import os
import time import time
basedir = os.getcwd() basedir = os.getcwd()
sys.path.extend(basedir + "/extensions/infinite-zoom-sd-webui/") sys.path.extend(basedir + "/extensions/infinite-zoom-automatic1111-webui/")
import numpy as np import numpy as np
import gradio as gr import gradio as gr
from PIL import Image from PIL import Image
@ -20,7 +20,7 @@ from modules.processing import (
from modules.ui import create_output_panel, plaintext_to_html from modules.ui import create_output_panel, plaintext_to_html
output_path = basedir + "/extensions/infinite-zoom-sd-webui/out" output_path = basedir + "/extensions/infinite-zoom-automatic1111-webui/out"
default_prompt = "A psychedelic jungle with trees that have glowing, fractal-like patterns, Simon stalenhag poster 1920s style, street level view, hyper futuristic, 8k resolution, hyper realistic" default_prompt = "A psychedelic jungle with trees that have glowing, fractal-like patterns, Simon stalenhag poster 1920s style, street level view, hyper futuristic, 8k resolution, hyper realistic"
default_negative_prompt = "frames, borderline, text, character, duplicate, error, out of frame, watermark, low quality, ugly, deformed, blur" default_negative_prompt = "frames, borderline, text, character, duplicate, error, out of frame, watermark, low quality, ugly, deformed, blur"
@ -218,8 +218,9 @@ def create_zoom(
save_path = os.path.join(output_path, "videos") save_path = os.path.join(output_path, "videos")
if not os.path.exists(save_path): if not os.path.exists(save_path):
os.makedirs(save_path) os.makedirs(save_path)
out = os.path.join(save_path, video_file_name)
write_video( write_video(
os.path.join(save_path, video_file_name), out,
all_frames, all_frames,
video_frame_rate, video_frame_rate,
video_zoom_mode, video_zoom_mode,
@ -228,7 +229,7 @@ def create_zoom(
) )
return ( return (
save_path, out,
processed.images, processed.images,
processed.js(), processed.js(),
plaintext_to_html(processed.info), plaintext_to_html(processed.info),