artegoser/image-pluser-webui
1
1
Fork 0
mirror of https://github.com/artegoser/image-pluser-webui.git synced 2024-10-05 14:45:58 +03:00
Code Issues Projects Releases Packages Wiki Activity

feat: video to images

Browse source
This commit is contained in:
Artemy 2023-04-16 10:12:04 +03:00
parent 7bf970cf1b
commit eb26452ad6
3 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
processing/bulk.py
View file

@ -18,7 +18,8 @@ def images_to_video(directory, fps, img_ext, img_name_format, video_name, video_
def video_to_images(video_path, img_ext): def video_to_images(video_path, img_ext):
images_pattern = generate_name( images_pattern = generate_name(
extension=img_ext, name=video_path, subfolder=os.path.join("images", get_date_text())) extension=img_ext, name=f"%d", subfolder=os.path.join("images", get_date_text()))
ffmpeg.input(video_path).output(images_pattern).run() ffmpeg.input(video_path).output(images_pattern).run()

4
processing/utils.py
View file

@ -4,12 +4,12 @@ import datetime
def generate_name(name=False, subfolder="stacked", extension="png", format='%Y-%m-%d_%H-%M-%S'): def generate_name(name=False, subfolder="stacked", extension="png", format='%Y-%m-%d_%H-%M-%S'):
os.makedirs(os.path.join("./output", subfolder), exist_ok=True) os.makedirs(os.path.join(".", "output", subfolder), exist_ok=True)
if name is False or name == "": if name is False or name == "":
name = get_date_text(format) name = get_date_text(format)
return os.path.join("./output", subfolder, f"{name}.{extension}") return os.path.join(".", "output", subfolder, f"{name}.{extension}")
def get_date_text(format='%Y-%m-%d_%H-%M-%S'): def get_date_text(format='%Y-%m-%d_%H-%M-%S'):

6
tabs/Video to images/app.py
View file

@ -1,12 +1,16 @@
import gradio as gr import gradio as gr
from processing.bulk import video_to_images
with gr.Blocks() as app: with gr.Blocks() as app:
gr.Markdown( gr.Markdown(
"Convert video to images. # **WIP, not working**") "Convert video to images.")
with gr.Row(): with gr.Row():
with gr.Column(): with gr.Column():
video = gr.Video(label="Video") video = gr.Video(label="Video")
with gr.Column():
format = gr.Radio( format = gr.Radio(
choices=["png", "jpg"], value="png", label="Format of image") choices=["png", "jpg"], value="png", label="Format of image")
submit = gr.Button("Submit") submit = gr.Button("Submit")
submit.click(fn=video_to_images, inputs=[video, format])