General Improvements

Author: ProGamerGov

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 @@ def INPUT_TYPES(s) -> Dict:
                 "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 @@ def INPUT_TYPES(s) -> Dict:
             "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 @@ def INPUT_TYPES(s) -> Dict:
                 "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 @@ def INPUT_TYPES(s) -> Dict:
                 "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 @@ def INPUT_TYPES(s) -> Dict:
                 "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"]