diff --git a/__init__.py b/__init__.py index dff9449..85fafc6 100644 --- a/__init__.py +++ b/__init__.py @@ -6,13 +6,13 @@ from .nodes import ( E2CNode, E2ENode, E2PNode, - MonoScopicToStereo, + MonoScopicToStereoNode, Pad180To360Node, PasteImageWithCoordsNode, RollImageNode, SplitFacesNode, StackFacesNode, - StereoToMonoScopic, + StereoToMonoScopicNode, ) NODE_CLASS_MAPPINGS = { @@ -28,8 +28,8 @@ NODE_CLASS_MAPPINGS = { "Paste Image with Coords": PasteImageWithCoordsNode, "Pad 180 to 360 Equirectangular": Pad180To360Node, "Crop 360 to 180 Equirectangular": Crop360To180Node, - "Crop Stereo to Monoscopic": StereoToMonoScopic, - "Merge Monoscopic into Stereo": MonoScopicToStereo, + "Crop Stereo to Monoscopic": StereoToMonoScopicNode, + "Merge Monoscopic into Stereo": StereoToMonoScopicNode, } NODE_DISPLAY_NAME_MAPPINGS = { diff --git a/nodes.py b/nodes.py index fcc2241..fdb3b76 100644 --- a/nodes.py +++ b/nodes.py @@ -564,7 +564,7 @@ class Crop360To180Node: return (cropped_img,) -class StereoToMonoScopic: +class StereoToMonoScopicNode: """ Split a stereo image into 2 monoscopic images. """ @@ -627,7 +627,7 @@ class StereoToMonoScopic: """ _, height, width, channels = image.shape - if split_direction == "horizontal": + if split_direction == "vertical": # Split horizontally (by height) mid_point = height // 2 if height % 2 != 0: # If height is odd @@ -641,7 +641,7 @@ class StereoToMonoScopic: # Even height, just split in the middle first_half = image[:, :mid_point, :, :] second_half = image[:, mid_point:, :] - elif split_direction == "vertical": + elif split_direction == "horizontal": # Split vertically (by width) mid_point = width // 2 if width % 2 != 0: # If width is odd @@ -665,7 +665,7 @@ class StereoToMonoScopic: return (first_half, second_half) -class MonoScopicToStereo: +class MonoScopicToStereoNode: """ Merge two monoscopic images into a stereo image. """ diff --git a/pyproject.toml b/pyproject.toml index f177097..b18ace8 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,7 +1,7 @@ [project] name = "comfyui-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 (virtual reality), HDRI environments (ex: skyboxes), image spheres, spherical images, 360 pano." -version = "1.0.2" +version = "1.0.3" license = {file = "LICENSE"} dependencies = ["pytorch360convert"]