EqualizeHistogram¶
Documentation¶
- Class name:
EqualizeHistogram
- Category:
Bmad/CV/Thresholding
- Output node:
False
This node applies histogram equalization to an image, enhancing its contrast by redistributing the image's intensity levels.
Input types¶
Required¶
src
- The source image to be processed. Histogram equalization will be applied to enhance its contrast.
- Comfy dtype:
IMAGE
- Python dtype:
torch.Tensor
Output types¶
image
- Comfy dtype:
IMAGE
- The processed image with enhanced contrast after applying histogram equalization.
- Python dtype:
torch.Tensor
- Comfy dtype:
Usage tips¶
- Infra type:
GPU
- Common nodes: unknown
Source code¶
class EqualizeHistogram:
@classmethod
def INPUT_TYPES(s):
return {
"required": {
"src": ("IMAGE",),
},
}
RETURN_TYPES = ("IMAGE",)
FUNCTION = "eq"
CATEGORY = "Bmad/CV/Thresholding"
def eq(self, src):
src = tensor2opencv(src, 1)
eq = cv.equalizeHist(src)
eq = cv.cvtColor(eq, cv.COLOR_GRAY2RGB)
eq = opencv2tensor(eq)
return (eq,)