From 656c321a60be01cb424e6cfc7fac8f387bb28d75 Mon Sep 17 00:00:00 2001 From: ProGamerGov Date: Thu, 27 Feb 2025 12:49:23 -0700 Subject: [PATCH] Fix bug in StereoToMonoScopic node --- __init__.py | 8 ++++---- nodes.py | 8 ++++---- pyproject.toml | 2 +- 3 files changed, 9 insertions(+), 9 deletions(-) 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"]