50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
from .utils import runwareUtils as rwUtils
|
|
|
|
|
|
class threeDInferenceInputs:
|
|
"""3D Inference Inputs node for configuring 3D generation inputs"""
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {},
|
|
"optional": {
|
|
"image": ("IMAGE", {
|
|
"tooltip": "Input image for 3D inference. This image will be used as the source for generating the 3D model."
|
|
}),
|
|
"mask": ("IMAGE", {
|
|
"tooltip": "Optional mask image to specify a region of interest in the input image for 3D generation."
|
|
}),
|
|
}
|
|
}
|
|
|
|
DESCRIPTION = "Configure custom inputs for Runware 3D Inference, including image and mask inputs."
|
|
FUNCTION = "createInputs"
|
|
RETURN_TYPES = ("RUNWARE3DINFERENCEINPUTS",)
|
|
RETURN_NAMES = ("Inputs",)
|
|
CATEGORY = "Runware"
|
|
|
|
def createInputs(self, **kwargs):
|
|
"""Create 3D inference inputs from provided parameters"""
|
|
image = kwargs.get("image", None)
|
|
mask = kwargs.get("mask", None)
|
|
|
|
inputs = {}
|
|
|
|
if image is not None:
|
|
inputs["image"] = rwUtils.convertTensor2IMG(image)
|
|
|
|
if mask is not None:
|
|
inputs["mask"] = rwUtils.convertTensor2IMG(mask)
|
|
|
|
return (inputs,)
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"Runware3DInferenceInputs": threeDInferenceInputs,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"Runware3DInferenceInputs": "Runware 3D Inference Inputs",
|
|
}
|