nodes/EllangoK-ComfyUI-post-processing-nodes
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Sepia modes over different channels

Browse Source
This commit is contained in:
EllangoK
2023-05-06 19:12:29 -04:00
parent 34dfeb316c
commit 5fdd0593ae
3 changed files with 19 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.gitignore
+1
View File
@@ -1,2 +1,3 @@
wip/
.vscode/*
*.pyc
post_processing/sepia.py
+9 -2
View File
@@ -15,6 +15,7 @@ class Sepia:
"max": 1.0,
"step": 0.1
}),
"mode": (["sepia", "blue-pia", "green-pia"],),
},
}
@@ -23,12 +24,18 @@ class Sepia:
CATEGORY = "postprocessing/Color Adjustments"
def sepia(self, image: torch.Tensor, strength: float):
def sepia(self, image: torch.Tensor, strength: float, mode: str = "sepia"):
if strength == 0:
return (image,)
sepia_weights = torch.tensor([0.2989, 0.5870, 0.1140]).view(1, 1, 1, 3).to(image.device)
sepia_filter = torch.tensor([1.0, 0.8, 0.6]).view(1, 1, 1, 3).to(image.device)
if mode == "sepia":
sepia_filter = torch.tensor([1.0, 0.8, 0.6]).view(1, 1, 1, 3).to(image.device)
elif mode == "blue-pia":
sepia_filter = torch.tensor([0.6, 0.8, 1.0]).view(1, 1, 1, 3).to(image.device)
elif mode == "green-pia":
sepia_filter = torch.tensor([0.6, 1.0, 0.6]).view(1, 1, 1, 3).to(image.device)
grayscale = torch.sum(image * sepia_weights, dim=-1, keepdim=True)
sepia = grayscale * sepia_filter
post_processing_nodes.py
+9 -2
View File
@@ -999,6 +999,7 @@ class Sepia:
"max": 1.0,
"step": 0.1
}),
"mode": (["sepia", "blue-pia", "green-pia"],),
},
}
@@ -1007,12 +1008,18 @@ class Sepia:
CATEGORY = "postprocessing/Color Adjustments"
def sepia(self, image: torch.Tensor, strength: float):
def sepia(self, image: torch.Tensor, strength: float, mode: str = "sepia"):
if strength == 0:
return (image,)
sepia_weights = torch.tensor([0.2989, 0.5870, 0.1140]).view(1, 1, 1, 3).to(image.device)
sepia_filter = torch.tensor([1.0, 0.8, 0.6]).view(1, 1, 1, 3).to(image.device)
if mode == "sepia":
sepia_filter = torch.tensor([1.0, 0.8, 0.6]).view(1, 1, 1, 3).to(image.device)
elif mode == "blue-pia":
sepia_filter = torch.tensor([0.6, 0.8, 1.0]).view(1, 1, 1, 3).to(image.device)
elif mode == "green-pia":
sepia_filter = torch.tensor([0.6, 1.0, 0.6]).view(1, 1, 1, 3).to(image.device)
grayscale = torch.sum(image * sepia_weights, dim=-1, keepdim=True)
sepia = grayscale * sepia_filter