diff --git a/nodes.py b/nodes.py index 3c5a70b..343c341 100644 --- a/nodes.py +++ b/nodes.py @@ -2733,6 +2733,28 @@ class GenerateNoise: noise = noise[0].repeat(batch_size, 1, 1, 1) return ({"samples":noise}, ) +class RemapDepth: + @classmethod + def INPUT_TYPES(s): + return {"required": { + "image": ("IMAGE",), + "min": ("FLOAT", {"default": 0.0,"min": -10.0, "max": 1.0, "step": 0.01}), + "max": ("FLOAT", {"default": 1.0,"min": 0.0, "max": 10.0, "step": 0.01}), + "clamp": ("BOOLEAN", {"default": True}), + }, + } + + RETURN_TYPES = ("IMAGE",) + FUNCTION = "remap" + + CATEGORY = "KJNodes/noise" + + def remap(self, image, min, max, clamp): + + image = min + image * (max - min) + if clamp: + image = torch.clamp(image, min=0.0, max=1.0) + return (image, ) NODE_CLASS_MAPPINGS = { "INTConstant": INTConstant, @@ -2785,7 +2807,8 @@ NODE_CLASS_MAPPINGS = { "AddLabel": AddLabel, "ReferenceOnlySimple3": ReferenceOnlySimple3, "SoundReactive": SoundReactive, - "GenerateNoise": GenerateNoise + "GenerateNoise": GenerateNoise, + "RemapDepth": RemapDepth } NODE_DISPLAY_NAME_MAPPINGS = { "INTConstant": "INT Constant", @@ -2837,6 +2860,7 @@ NODE_DISPLAY_NAME_MAPPINGS = { "AddLabel": "AddLabel", "ReferenceOnlySimple3": "ReferenceOnlySimple3", "SoundReactive": "SoundReactive", - "GenerateNoise": "GenerateNoise" + "GenerateNoise": "GenerateNoise", + "RemapDepth": "RemapDepth" } \ No newline at end of file