Adds a custom ComfyUI node (GaussianSplattingPreview) that accepts two input images and outputs a novel-view preview rendered with 3D Gaussian Splatting. Pipeline: SIFT/ORB feature matching → essential matrix → triangulation → isotropic Gaussian initialization → PyTorch-based batched splatting renderer. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
20 lines
526 B
Python
20 lines
526 B
Python
try:
|
|
from .gs_preview_node import GaussianSplattingPreviewNode
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"GaussianSplattingPreview": GaussianSplattingPreviewNode,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"GaussianSplattingPreview": "3DGS Preview (2 Images)",
|
|
}
|
|
|
|
print("[ComfyUI-3DGS] Loaded successfully.")
|
|
|
|
except Exception as e:
|
|
print(f"[ComfyUI-3DGS] Failed to load: {e}")
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|