perf: auto generated list of methods

This commit is contained in:
Artemy 2023-04-21 14:25:22 +03:00
parent 0adccad87b
commit 8fe6d70749
3 changed files with 5 additions and 7 deletions

View file

@ -2,8 +2,6 @@ from PIL import Image, ImageFilter
from skimage import feature from skimage import feature
import numpy as np import numpy as np
methods = ["Sharpen", "Edge detection", "Canny edge detection"]
def edge(file_name): def edge(file_name):
image = Image.open(file_name) image = Image.open(file_name)
@ -31,3 +29,5 @@ methods_funcs = {
"Edge detection": edge, "Edge detection": edge,
"Canny edge detection": canny_edge, "Canny edge detection": canny_edge,
} }
methods = list(methods_funcs.keys())

View file

@ -2,8 +2,6 @@ from PIL import Image, ImageChops
from tqdm import tqdm from tqdm import tqdm
import numpy as np import numpy as np
methods = ["Denoise", "StarTracks", "Noise extractor", "Untrack"]
def denoise(files): def denoise(files):
images = [np.asarray(Image.open(file)) for file in tqdm(files)] images = [np.asarray(Image.open(file)) for file in tqdm(files)]
@ -46,3 +44,5 @@ methods_funcs = {
"Noise extractor": noise_extractor, "Noise extractor": noise_extractor,
"Untrack": untrack, "Untrack": untrack,
} }
methods = list(methods_funcs.keys())

View file

@ -3,9 +3,7 @@ from processing.bulk import bulk_processing
from methods.bulk_methods import methods from methods.bulk_methods import methods
with gr.Blocks() as app: with gr.Blocks() as app:
gr.Markdown( gr.Markdown("Mass processing of images one at a time.")
"Mass processing of images one at a time and saving to video if needed. # **WIP, not working**"
)
with gr.Row(): with gr.Row():
with gr.Column(): with gr.Column():
directory = gr.Text( directory = gr.Text(