mirror of
https://github.com/artegoser/image-pluser-webui.git
synced 2025-02-22 20:23:15 +03:00
feat: some not working tabs
This commit is contained in:
parent
f338f7b5c9
commit
efcd1ccbac
4 changed files with 128 additions and 90 deletions
30
processing/stacking.py
Normal file
30
processing/stacking.py
Normal file
|
@ -0,0 +1,30 @@
|
|||
from methods.stack_methods import denoise, startracks, noise_extractor, untrack
|
||||
import os
|
||||
import datetime
|
||||
|
||||
|
||||
def generate_name():
|
||||
os.makedirs("./output/stacked", exist_ok=True)
|
||||
|
||||
return f"./output/stacked/{datetime.datetime.now().strftime('%Y-%m-%d_%H-%M-%S')}.png"
|
||||
|
||||
|
||||
def stacking(dir, method):
|
||||
|
||||
files = os.listdir(dir)
|
||||
files = list(map(lambda x: os.path.join(dir, x), files))
|
||||
files = list(filter(lambda x: x.endswith(".png"), files))
|
||||
|
||||
if method == "denoise":
|
||||
img = denoise(files)
|
||||
elif method == "startracks":
|
||||
img = startracks(files)
|
||||
elif method == "noise extractor":
|
||||
img = noise_extractor(files)
|
||||
elif method == "untrack":
|
||||
img = untrack(files)
|
||||
|
||||
name = generate_name()
|
||||
img.save(name)
|
||||
|
||||
return [name]
|
Loading…
Add table
Reference in a new issue