wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-27 17:37:39 +08:00
Code Issues Actions 17 Packages Projects Releases Wiki Activity

Slice alpha channel

Browse Source
This commit is contained in:
kijai 2026-05-14 12:52:16 +03:00
parent 10837132a4
commit 7a5aff1946
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy_extras/nodes_moge.py
View File

@ -141,6 +141,7 @@ class MoGePanoramaInference(io.ComfyNode):
if image.shape[0] != 1: if image.shape[0] != 1:
raise ValueError(f"MoGePanoramaInference takes a single image (got batch of {image.shape[0]})") raise ValueError(f"MoGePanoramaInference takes a single image (got batch of {image.shape[0]})")
image = image[..., :3]
H, W = int(image.shape[1]), int(image.shape[2]) H, W = int(image.shape[1]), int(image.shape[2])
scale = min(merge_resolution / max(H, W), 1.0) scale = min(merge_resolution / max(H, W), 1.0)
merge_h, merge_w = max(int(H * scale), 32), max(int(W * scale), 32) merge_h, merge_w = max(int(H * scale), 32), max(int(W * scale), 32)
@ -257,6 +258,7 @@ class MoGeInference(io.ComfyNode):
@classmethod @classmethod
def execute(cls, moge_model, image, resolution_level, fov_x_degrees, batch_size, force_projection, apply_mask) -> io.NodeOutput: def execute(cls, moge_model, image, resolution_level, fov_x_degrees, batch_size, force_projection, apply_mask) -> io.NodeOutput:
image = image[..., :3]
bchw = image.movedim(-1, -3).contiguous() bchw = image.movedim(-1, -3).contiguous()
B = bchw.shape[0] B = bchw.shape[0]
fov = None if fov_x_degrees <= 0 else float(fov_x_degrees) fov = None if fov_x_degrees <= 0 else float(fov_x_degrees)