Fix bug in StereoToMonoScopic node
This commit is contained in:
+4
-4
@@ -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 = {
|
||||
|
||||
@@ -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
@@ -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"]
|
||||
|
||||
|
||||
Reference in New Issue
Block a user