Add 3D infrence support

2026-01-29 21:51:35 -05:00
parent 938185cd42
commit ff6b7d8ad4
5 changed files with 357 additions and 0 deletions
@@ -0,0 +1,49 @@
+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",
+}