Spaces:
Runtime error
Runtime error
File size: 2,200 Bytes
f2dbf59 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
from .imagefunc import *
NODE_NAME = 'ChannelShake'
class ChannelShake:
def __init__(self):
pass
@classmethod
def INPUT_TYPES(self):
channel_mode = ['RGB', 'RBG', 'BGR', 'BRG', 'GBR', 'GRB']
return {
"required": {
"image": ("IMAGE", ), #
"distance": ("INT", {"default": 20, "min": 1, "max": 999, "step": 1}), # 距离
"angle": ("FLOAT", {"default": 40, "min": -360, "max": 360, "step": 0.1}), # 角度
"mode": (channel_mode,), # 模式
},
"optional": {
}
}
RETURN_TYPES = ("IMAGE",)
RETURN_NAMES = ("image",)
FUNCTION = 'channel_shake'
CATEGORY = '😺dzNodes/LayerFilter'
def channel_shake(self, image, distance, angle, mode, ):
ret_images = []
for i in image:
i = torch.unsqueeze(i, 0)
_canvas = tensor2pil(i).convert('RGB')
R, G, B = _canvas.split()
x = int(math.cos(angle) * distance)
y = int(math.sin(angle) * distance)
if mode.startswith('R'):
R = shift_image(R.convert('RGB'), -x, -y).convert('L')
if mode.startswith('G'):
G = shift_image(G.convert('RGB'), -x, -y).convert('L')
if mode.startswith('B'):
B = shift_image(B.convert('RGB'), -x, -y).convert('L')
if mode.endswith('R'):
R = shift_image(R.convert('RGB'), x, y).convert('L')
if mode.endswith('G'):
G = shift_image(G.convert('RGB'), x, y).convert('L')
if mode.endswith('B'):
B = shift_image(B.convert('RGB'), x, y).convert('L')
ret_image = Image.merge('RGB', [R, G, B])
ret_images.append(pil2tensor(ret_image))
log(f"{NODE_NAME} Processed {len(ret_images)} image(s).", message_type='finish')
return (torch.cat(ret_images, dim=0),)
NODE_CLASS_MAPPINGS = {
"LayerFilter: ChannelShake": ChannelShake
}
NODE_DISPLAY_NAME_MAPPINGS = {
"LayerFilter: ChannelShake": "LayerFilter: ChannelShake"
} |