feat: some not working tabs

This commit is contained in:
Artemy 2023-04-15 19:36:31 +03:00
parent f338f7b5c9
commit efcd1ccbac
4 changed files with 128 additions and 90 deletions

30
processing/stacking.py Normal file
View 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]