General Improvements

ReadMe.md

@@ -52,6 +52,12 @@ Or for Windows:
 path/to/python.exe -m pip install pytorch360convert
 ```
 
+#### ComfyUI registry install
+
+```
+comfy node registry-install pytorch360convert
+```
+
 ## 💡 Usage Examples
 
 See the provided workflow files in the `examples` directory for examples of how to use each new node.

nodes.py

@@ -16,8 +16,8 @@ class C2ENode:
                 "e_img": ("IMAGE", {"default": None}),
                 "h": ("INT", {"default": -1}),
                 "w": ("INT", {"default": -1}),
-                "padding_mode": ("STRING", {"default": "bilinear"}),
-                "cube_format": ("STRING", {"default": "stack"}),
+                "padding_mode": (["bilinear", "bicubic", "nearest"], {"default": "bilinear"}),
+                "cube_format": (["stack", "dice", "horizon", "list", "dict"], {"default": "stack"}),
             },
         }
 
@@ -61,8 +61,8 @@ class E2CNode:
             "required": {
                 "e_img": ("IMAGE", {"default": None}),
                 "face_width": ("INT", {"default": -1}),
-                "padding_mode": ("STRING", {"default": "bilinear"}),
-                "cube_format": ("STRING", {"default": "stack"}),
+                "padding_mode": (["bilinear", "bicubic", "nearest"], {"default": "bilinear"}),
+                "cube_format": (["stack", "dice", "horizon", "list", "dict"], {"default": "stack"}),
             },
         }
 
@@ -112,7 +112,7 @@ class E2PNode:
                 "out_h": ("INT", {"default": 512}),
                 "out_w": ("INT", {"default": 512}),
                 "in_rot_deg": ("FLOAT", {"default": 0.0}),
-                "padding_mode": ("STRING", {"default": "bilinear"}),
+                "padding_mode": (["bilinear", "bicubic", "nearest"], {"default": "bilinear"}),
             },
         }
 
@@ -163,7 +163,7 @@ class E2ENode:
                 "roll": ("FLOAT", {"default": 0.0}),
                 "h_deg": ("FLOAT", {"default": 0.0}),
                 "v_deg": ("FLOAT", {"default": 0.0}),
-                "padding_mode": ("STRING", {"default": "bilinear"}),
+                "padding_mode": (["bilinear", "bicubic", "nearest"], {"default": "bilinear"}),
             },
         }
 
@@ -312,8 +312,8 @@ class C2EMaskedDiffNode:
                 "original_faces": ("IMAGE", {"default": None}),
                 "modified_faces": ("IMAGE", {"default": None}),
                 "original_equi": ("IMAGE", {"default": None}),
-                "padding_mode": ("STRING", {"default": "bilinear"}),
-                "cube_format": ("STRING", {"default": "stack"}),
+                "padding_mode": (["bilinear", "bicubic", "nearest"], {"default": "bilinear"}),
+                "cube_format": (["stack", "dice", "horizon", "list", "dict"], {"default": "stack"}),
             },
         }
 

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "pytorch360convert"
 description = "A collection of custom nodes for working with and converting between 360 degree equirectangular images, cubemap, and perspective images. Panoramic 360 images are also sometimes known as 'VR photography', 'image sphere', and 360 pano."
-version = "1.0.0"
+version = "1.0.1"
 license = {file = "LICENSE"}
 dependencies = ["pytorch360convert"]