Spaces:
Running
on
Zero
Running
on
Zero
import nodes | |
import folder_paths | |
MAX_RESOLUTION = nodes.MAX_RESOLUTION | |
class WebcamCapture(nodes.LoadImage): | |
def INPUT_TYPES(s): | |
return { | |
"required": { | |
"image": ("WEBCAM", {}), | |
"width": ("INT", {"default": 0, "min": 0, "max": MAX_RESOLUTION, "step": 1}), | |
"height": ("INT", {"default": 0, "min": 0, "max": MAX_RESOLUTION, "step": 1}), | |
"capture_on_queue": ("BOOLEAN", {"default": True}), | |
} | |
} | |
RETURN_TYPES = ("IMAGE",) | |
FUNCTION = "load_capture" | |
CATEGORY = "image" | |
def load_capture(s, image, **kwargs): | |
return super().load_image(folder_paths.get_annotated_filepath(image)) | |
NODE_CLASS_MAPPINGS = { | |
"WebcamCapture": WebcamCapture, | |
} | |
NODE_DISPLAY_NAME_MAPPINGS = { | |
"WebcamCapture": "Webcam Capture", | |
} |