Fix bug in StereoToMonoScopic node

This commit is contained in:
ProGamerGov
2025-02-27 12:49:23 -07:00
committed by GitHub
parent bc77b1f59c
commit 656c321a60
3 changed files with 9 additions and 9 deletions
+4 -4
View File
@@ -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 = {
+4 -4
View File
@@ -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.
"""
+1 -1
View File
@@ -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"]