File size: 725 Bytes
c10aebf |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
from PIL import Image
from sd_bmab import util
from sd_bmab.base.context import Context
from sd_bmab.base.processorbase import ProcessorBase
class NoiseAlpha(ProcessorBase):
def __init__(self) -> None:
super().__init__()
self.noise_alpha = 0
def preprocess(self, context: Context, image: Image):
self.noise_alpha = context.args['noise_alpha']
return self.noise_alpha != 0
def process(self, context: Context, image: Image):
context.add_generation_param('BMAB noise alpha', self.noise_alpha)
img_noise = util.generate_noise(context.sdprocessing.seed, image.size[0], image.size[1])
return Image.blend(image, img_noise, alpha=self.noise_alpha)
def postprocess(self, context: Context, image: Image):
pass
|