artegoser/image-pluser-webui
1
1
Fork 0
mirror of https://github.com/artegoser/image-pluser-webui.git synced 2024-11-21 19:36:21 +03:00
Code Issues Projects Releases Packages Wiki Activity

fix: tqdm for bulk

Browse source
This commit is contained in:
Artemy 2023-04-18 13:21:35 +03:00
parent 09346cb114
commit 123d986f47
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
processing/bulk.py
View file

@ -2,6 +2,7 @@ import ffmpeg
import os import os
from processing.utils import generate_name, get_date_text, generate_name_with_file_name from processing.utils import generate_name, get_date_text, generate_name_with_file_name
from methods.bulk_methods import methods_funcs from methods.bulk_methods import methods_funcs
from tqdm import tqdm
def images_to_video( def images_to_video(
@ -30,12 +31,11 @@ def video_to_images(video_path, img_ext):
def bulk_processing(directory, out_directory, method): def bulk_processing(directory, out_directory, method):
date = get_date_text() run_bulk(methods_funcs[method], directory, out_directory, get_date_text())
run_bulk(methods_funcs[method], directory, out_directory, date)
def run_bulk(func, directory, out_directory, date): def run_bulk(func, directory, out_directory, date):
for file in os.listdir(directory): for file in tqdm(os.listdir(directory)):
img = func(os.path.join(directory, file)) img = func(os.path.join(directory, file))
if out_directory: if out_directory:
img_out_path = os.path.join(out_directory, file) img_out_path = os.path.join(out_directory, file)