Compare commits

..
Author SHA1 Message Date
google-labs-jules[bot] d49f3f03ce Fix readability 2025-05-22 20:18:58 +00:00
16 changed files with 479 additions and 3709 deletions
-19
View File
@@ -1,19 +0,0 @@
.PHONY: install install_all install_modules download_flux download_vae
# install everything except WAN‑VACE downloads
install:
./install.sh install
# install everything + WAN‑VACE + HF login
install_all:
./install.sh all
# lower‑level helpers
install_modules:
./install.sh modules
download_flux:
./install.sh flux
download_vae:
./install.sh vae
+7 -59
View File
@@ -1,5 +1,5 @@
# camera-comfyUI
[![Ask DeepWiki](https://deepwiki.com/badge.svg)](https://deepwiki.com/Alexankharin/camera-comfyUI)
![ComfyUI Custom Nodes](demo_images/Camera_interpolation_pointcloud.gif)
> Custom ComfyUI nodes for advanced reprojections, point cloud processing, and camera-driven workflows.
@@ -80,25 +80,18 @@ A collection of ComfyUI custom nodes to handle diverse camera projections (pinho
* ### Reprojection Nodes
* `ReprojectImage`, `ReprojectDepth`, `OutpaintAnyProjection`
* ### Matrix Nodes
* `TransformToMatrix`, `TransformToMatrixManual`
* ### Depth Nodes
* `DepthEstimatorNode`, `DepthToImageNode`, `ZDepthToRayDepthNode`
* `CombineDepthsNode`, `DepthRenormalizer`, `FisheyeDepthEstimator`
* `CombineDepthsNode`, `DepthRenormalizer`
* ### Point Cloud Nodes
* `DepthToPointCloud`, `TransformPointCloud`, `ProjectPointCloud`, `PointCloudUnion`
* `PointCloudCleaner`, `LoadPointCloud`, `SavePointCloud`, `ProjectAndClean`
* ### Trajectory Nodes
* `DepthToPointCloud`, `TransformPointCloud`, `ProjectPointCloud`
* `PointCloudUnion`, `PointCloudCleaner`, `LoadPointCloud`, `SavePointCloud`
* `CameraMotionNode`, `CameraInterpolationNode`, `CameraTrajectoryNode`
* `SaveTrajectory`, `LoadTrajectory`, `PointcloudTrajectoryEnricher`
---
@@ -117,7 +110,7 @@ A collection of ComfyUI custom nodes to handle diverse camera projections (pinho
| `ZDepthToRayDepthNode` | Converts Z-depth (output of metric-depth-anything) to ray depth to compensate lens curvature. |
| `TransformPointCloud` | Applies 4×4 rotation matrix to point cloud |
| `ProjectPointCloud` | Z-buffer–based projection of point cloud into image + mask. |
| `CameraMotionNode` | Generates image and mask sequences along a camera trajectory with optional mask dilation/inversion. |
| `CameraMotionNode` | Generates image sequences by moving camera along a trajectory. |
| `CameraInterpolationNode` | Builds a trajectory tensor from two poses. |
| `CameraTrajectoryNode` | Interactive Open3D GUI for recording camera waypoints. |
| `PointCloudCleaner` | Removes isolated points via voxel filtering. |
@@ -139,7 +132,6 @@ A set of JSON workflows illustrating typical use cases. Each workflow lives in `
| **Pointcloud.json** | Metric‐depth‐anything v2 → point cloud → camera view synthesis |
| **pointcloud\_inpaint.json** | Inpaint + backproject to 3D for dynamic camera motion videos |
| **Pointcloud\_walker.json** | GUI‐based camera control via Open3D |
| **sbs180\_workflow.json** | Generate stereo (side-by-side) wide-angle/fisheye/equirectangular stereo pairs from a high-res input |
---
@@ -198,53 +190,10 @@ Inpaint image with shifted camera and backproject for dynamic camera‐driven vi
<img src="demo_images/Fisheye_camera_pointcloud_moved_outpainted.png" alt="PointCloud Inpaint" width="40%" />
<img src="demo_images/Camera_interpolation_pointcloud.gif" alt="PointCloud Inpaint Video" width="40%" />
### 9. `sbs180_workflow.json`
Take a wide-angle (fisheye or equirectangular) high-resolution (e.g., 4096×4096) image and generate a stereo pair by moving the camera horizontally. The output is a wide-angle stereo pair (side-by-side), simulating a fisheye or equirectangular stereo camera.
<img src="demo_images/equirect_stereo.gif" alt="Equirectangular Stereo Demo" width="80%" />
### 10. `Pointcloud_walker.json`
Interactive Open3D-based GUI for walking and setting camera trajectory inside pointcloud.
### 11. `wan-vace_ref_to_video.json`
Integrate the [wan2.1-vace] video generation model to inpaint empty or newly revealed regions during camera movement or view synthesis. This workflow demonstrates how to use the camera-comfyUI nodes to generate camera trajectories and masks, then fill missing areas with the video inpainting model for smooth, high-quality results.
<img src="demo_images/wan-vace-camera.gif" alt="wan2.1-vace Camera Inpainting Demo" width="80%" />
---
## Trajectory Concept
A **trajectory** in camera-comfyUI is a sequence of camera poses, each represented as a 4×4 transformation matrix. This set of matrices defines the path and orientation of the camera through 3D space, enabling smooth and complex camera movements for view synthesis, point cloud rendering, and video generation.
### Creating Trajectories
There are two main ways to create a trajectory:
- **Camera Matrices Interpolation:**
Define two or more camera poses (as matrices), and interpolate between them to generate a smooth path. The `CameraInterpolationNode` automates this process, producing a trajectory tensor for use in camera motion nodes.
- **Walking in Open3D Environment:**
Use the interactive Open3D GUI (`CameraTrajectoryNode`) to "walk" through the point cloud. As you move the camera, waypoints (poses) are recorded, forming a trajectory that can be exported and reused.
### Using Trajectories
The `CameraMotionNode` takes a trajectory (set of matrices) and interpolates camera positions and orientations along it, producing smooth camera movements for rendering sequences or videos.
---
## Point Cloud Formats
Point clouds can be saved and loaded in two formats:
- **.npy**: Numpy array format (fast, preserves all tensor data, recommended for internal pipelines).
- **.ply**: Polygon File Format (widely supported, viewable in external 3D tools).
Use the `SavePointCloud` and `LoadPointCloud` nodes to handle I/O operations in either format.
---
## Contributing
@@ -253,12 +202,11 @@ Contributions welcome! Please open issues or PRs to add features, improve docs,
## TODO List
* [x] Add processing to pointcloud or depthmap to remove outlier and lonely points at depth borders.
* [ ] Add processing to pointcloud or depthmap to remove outlier and lonely points at depth borders.
* [x] Use built-in comfyUI mask type an image.
* [x] Unite nodes into groups to simplify workflows.
* [ ] Create a single workflow for view synthesis.
* [x] Implement easier and more flexible camera control - more complex camera movements with more than 2 points.
* [x] Add more examples and documentation for each node.
* [x] Add pointcloud union
* [x] Fix imports for renamed folders (e.g., inpainting_flux)
* [x] Integrate camera movement pipeline with video models (e.g., wan2.1) for smooth, high-quality inpainting along camera trajectories.
* [ ] Fix imports for renamed folders (e.g., inpainting_flux)
+1 -2
View File
@@ -3,7 +3,6 @@ from .reprojection_nodes import NODE_CLASS_MAPPINGS as NCM2
from .metric_depth_nodes import NODE_CLASS_MAPPINGS as NCM3
from .flux_fisheye_filling_nodes import NODE_CLASS_MAPPINGS as NCM4
from .complex_nodes import NODE_CLASS_MAPPINGS as NCM5
from .video_nodes import NODE_CLASS_MAPPINGS as NCM6
NODE_CLASS_MAPPINGS = {**NCM1, **NCM2, **NCM3, **NCM4, **NCM5, **NCM6}
NODE_CLASS_MAPPINGS = {**NCM1, **NCM2, **NCM3, **NCM4, **NCM5}
__all__ = ["NODE_CLASS_MAPPINGS"]
+376 -168
View File
@@ -64,7 +64,7 @@ class FisheyeDepthEstimator:
RETURN_TYPES = ("TENSOR","MASK")
RETURN_NAMES = ("depthmap","mask")
FUNCTION = "estimate_fisheye_depth"
CATEGORY = "Camera/Depth"
CATEGORY = "Camera/depth"
def estimate_fisheye_depth(
self,
@@ -95,76 +95,46 @@ class FisheyeDepthEstimator:
depth_full, = de_node.estimate_depth(image, model_name, depth_scale)
mask_full = (depth_full > 0).float()
# 2) Pinhole orientations (5 views)
rotations = [
(0, 0, 0), # front
(0, 45, 0), # right
(0, -45, 0), # left
(45, 0, 0), # up
(-45, 0, 0), # down
]
# 2) Generate Pinhole Views
fisheye_depths, fisheye_masks = self._generate_pinhole_views(
image,
de_node, z2r_node, ri_node, rd_node,
fisheye_fov, pinhole_fov,
pin_w, pin_h, fish_w, fish_h,
model_name, depth_scale, median_blur_kernel
)
fisheye_depths = []
fisheye_masks = []
# euler → matrix
def euler_to_matrix(pitch, yaw, roll):
p, y, r = map(math.radians, (pitch, yaw, roll))
Rx = torch.tensor([[1,0,0],[0,math.cos(p),-math.sin(p)],[0,math.sin(p),math.cos(p)]], dtype=torch.float32)
Ry = torch.tensor([[math.cos(y),0,math.sin(y)],[0,1,0],[-math.sin(y),0,math.cos(y)]], dtype=torch.float32)
Rz = torch.tensor([[math.cos(r),-math.sin(r),0],[math.sin(r),math.cos(r),0],[0,0,1]], dtype=torch.float32)
R = Rz @ Ry @ Rx
M = torch.eye(4, dtype=torch.float32)
M[:3, :3] = R
return M
# 3) Process each orientation
for pitch, yaw, roll in rotations:
M = euler_to_matrix(pitch, yaw, roll)
M_np = M.numpy()
M_inv = torch.inverse(M).numpy()
# fisheye → pinhole
img_pin, mask_pin = ri_node.reproject_image(
image,
input_horiszontal_fov = fisheye_fov,
output_horiszontal_fov= pinhole_fov,
input_projection = "FISHEYE",
output_projection = "PINHOLE",
output_width = pin_w,
output_height = pin_h,
transform_matrix = M_np,
feathering = 0,
)
# estimate pinhole depth
depth_pin, = de_node.estimate_depth(img_pin, model_name, depth_scale, median_blur_kernel=median_blur_kernel)
depth_pin, = z2r_node.depth_to_ray_depth(
depth_pin,
pinhole_fov,
)
# pinhole → fisheye
fish_depth, fish_mask = rd_node.reproject_depth(
depth_pin,
input_horizontal_fov = pinhole_fov,
output_horizontal_fov= fisheye_fov,
input_projection = "PINHOLE",
output_projection = "FISHEYE",
output_width = fish_w,
output_height = fish_h,
transform_matrix = M_inv,
)
# squeeze mask to [B,H,W]
fish_mask = fish_mask.squeeze(1)
fisheye_depths.append(fish_depth) # [B,H,W]
fisheye_masks.append(fish_mask)
fisheye_depths.append(depth_full) # [B,H,W]
fisheye_masks.append(mask_full.squeeze(-1)) # [B,H,W 1]
# merged mask
merged_mask = torch.sum(torch.stack(fisheye_masks), dim=0) > 0.5
# print(fisheye_depths[0].shape, fisheye_depths[-1].shape, merged_mask.shape)
# 4) Merge in sequence
d_acc, m_acc = self._merge_depths(
fisheye_depths, fisheye_masks,
ren_node, comb_node,
mode, softmerge_radius
)
# 5) Circular mask
ys = torch.arange(fish_h, device=d_acc.device).view(1, fish_h, 1)
xs = torch.arange(fish_w, device=d_acc.device).view(1, 1, fish_w)
cy = (fish_h - 1) / 2.0
cx = (fish_w - 1) / 2.0
dist2 = (ys - cy)**2 + (xs - cx)**2
radius2 = (min(fish_w, fish_h) / 2.0)**2
circ_mask = (dist2 <= radius2).float()
return d_acc, circ_mask
def _merge_depths(
self,
fisheye_depths: list,
fisheye_masks: list,
ren_node: DepthRenormalizer,
comb_node: CombineDepthsNode,
mode: str,
softmerge_radius: int,
) -> Tuple[torch.Tensor, torch.Tensor]:
d_acc = fisheye_depths[0]
m_acc = fisheye_masks[0]
for d_new, m_new in zip(fisheye_depths[1:-1], fisheye_masks[1:-1]):
@@ -187,20 +157,120 @@ class FisheyeDepthEstimator:
m_acc,
d_norm,
m_new_last,
mode = "SRC",
mode = "SRC", # Use SRC for the full fisheye to preserve its details
invert_mask = False,
softmerge_radius = softmerge_radius
)
return d_acc, m_acc
# 5) Circular mask
ys = torch.arange(fish_h, device=d_acc.device).view(1, fish_h, 1)
xs = torch.arange(fish_w, device=d_acc.device).view(1, 1, fish_w)
cy = (fish_h - 1) / 2.0
cx = (fish_w - 1) / 2.0
dist2 = (ys - cy)**2 + (xs - cx)**2
radius2 = (min(fish_w, fish_h) / 2.0)**2
circ_mask = (dist2 <= radius2).float()
return d_acc, circ_mask
def _generate_pinhole_views(
self,
image: torch.Tensor,
de_node: DepthEstimatorNode,
z2r_node: ZDepthToRayDepthNode,
ri_node: ReprojectImage,
rd_node: ReprojectDepth,
fisheye_fov: float,
pinhole_fov: float,
pin_w: int,
pin_h: int,
fish_w: int,
fish_h: int,
model_name: str,
depth_scale: float,
median_blur_kernel: int,
) -> Tuple[list, list]:
rotations = [
(0, 0, 0), # front
(0, 45, 0), # right
(0, -45, 0), # left
(45, 0, 0), # up
(-45, 0, 0), # down
]
fisheye_depths = []
fisheye_masks = []
for pitch, yaw, roll in rotations:
M = self._euler_to_matrix(pitch, yaw, roll)
M_np = M.numpy()
M_inv = torch.inverse(M).numpy()
fish_depth, fish_mask = self._process_view(
image, M_np, M_inv,
de_node, z2r_node, ri_node, rd_node,
fisheye_fov, pinhole_fov,
pin_w, pin_h, fish_w, fish_h,
model_name, depth_scale, median_blur_kernel
)
fisheye_depths.append(fish_depth)
fisheye_masks.append(fish_mask)
return fisheye_depths, fisheye_masks
def _process_view(
self,
image: torch.Tensor,
M_np: np.ndarray,
M_inv: np.ndarray,
de_node: DepthEstimatorNode,
z2r_node: ZDepthToRayDepthNode,
ri_node: ReprojectImage,
rd_node: ReprojectDepth,
fisheye_fov: float,
pinhole_fov: float,
pin_w: int,
pin_h: int,
fish_w: int,
fish_h: int,
model_name: str,
depth_scale: float,
median_blur_kernel: int,
) -> Tuple[torch.Tensor, torch.Tensor]:
# fisheye → pinhole
img_pin, mask_pin = ri_node.reproject_image(
image,
input_horiszontal_fov = fisheye_fov,
output_horiszontal_fov= pinhole_fov,
input_projection = "FISHEYE",
output_projection = "PINHOLE",
output_width = pin_w,
output_height = pin_h,
transform_matrix = M_np,
feathering = 0,
)
# estimate pinhole depth
depth_pin, = de_node.estimate_depth(img_pin, model_name, depth_scale, median_blur_kernel=median_blur_kernel)
depth_pin, = z2r_node.depth_to_ray_depth(
depth_pin,
pinhole_fov,
)
# pinhole → fisheye
fish_depth, fish_mask = rd_node.reproject_depth(
depth_pin,
input_horizontal_fov = pinhole_fov,
output_horizontal_fov= fisheye_fov,
input_projection = "PINHOLE",
output_projection = "FISHEYE",
output_width = fish_w,
output_height = fish_h,
transform_matrix = M_inv,
)
# squeeze mask to [B,H,W]
fish_mask = fish_mask.squeeze(1)
return fish_depth, fish_mask
def _euler_to_matrix(self, pitch, yaw, roll):
p, y, r = map(math.radians, (pitch, yaw, roll))
Rx = torch.tensor([[1,0,0],[0,math.cos(p),-math.sin(p)],[0,math.sin(p),math.cos(p)]], dtype=torch.float32)
Ry = torch.tensor([[math.cos(y),0,math.sin(y)],[0,1,0],[-math.sin(y),0,math.cos(y)]], dtype=torch.float32)
Rz = torch.tensor([[math.cos(r),-math.sin(r),0],[math.sin(r),math.cos(r),0],[0,0,1]], dtype=torch.float32)
R = Rz @ Ry @ Rx
M = torch.eye(4, dtype=torch.float32)
M[:3, :3] = R
return M
class PointcloudTrajectoryEnricher:
"""
@@ -241,7 +311,7 @@ class PointcloudTrajectoryEnricher:
RETURN_TYPES = ("TENSOR","IMAGE","TENSOR")
RETURN_NAMES = ("enriched_pointcloud","debug_image","debug_depth")
FUNCTION = "enrich_trajectory"
CATEGORY = "Camera/Trajectory"
CATEGORY = "Camera/pointcloud"
def enrich_trajectory(
self,
@@ -288,106 +358,244 @@ class PointcloudTrajectoryEnricher:
debug_img = torch.zeros((1, height, width, 3), device=device)
debug_depth = torch.zeros((1, height, width, 1), device=device)
enriched_pc = pointcloud
# Initialize debug_img and debug_depth which will be updated in the loop
# and will hold the values from the last processed view.
debug_img = torch.zeros((1, height, width, 3), device=device)
debug_depth = torch.zeros((1, height, width, 1), device=device)
# loop over trajectory (limit or full)
for M in tqdm(trajectory[:15], desc="Enriching trajectory"):
M_np = M.cpu().numpy()
M_inv = np.linalg.inv(M_np)
enriched_pc, view_debug_img, view_debug_depth = self._process_single_view(
M, enriched_pc, device,
proj_node, outpaint_node, depth_node, renorm_node,
depth2pc_node, transform_node, clean_node, zdepth_node,
camera_type, horizontal_fov, width, height,
patch_projection, patch_horiz_fov, patch_res,
patch_phi, patch_theta, prompt,
num_inference_steps, guidance_scale, mask_blur,
voxel_size, min_points_per_voxel, model_name
)
debug_img = view_debug_img
debug_depth = view_debug_depth
return enriched_pc, debug_img, debug_depth
# transform and select front points
rotated, = transform_node.transform_pointcloud(enriched_pc, M_np)
pc_front = rotated[rotated[:, 2] > 0]
def _process_single_view(
self,
M_matrix: torch.Tensor,
current_enriched_pc: torch.Tensor,
device: torch.device,
proj_node: ProjectPointCloud,
outpaint_node: OutpaintAnyProjection,
depth_node: DepthEstimatorNode,
renorm_node: DepthRenormalizer,
depth2pc_node: DepthToPointCloud,
transform_node: TransformPointCloud,
clean_node: PointCloudCleaner,
zdepth_node: ZDepthToRayDepthNode,
camera_type: str,
horizontal_fov: float,
width: int,
height: int,
patch_projection: str,
patch_horiz_fov: float,
patch_res: int,
patch_phi: float,
patch_theta: float,
prompt: str,
num_inference_steps: int,
guidance_scale: float,
mask_blur: int,
voxel_size: float,
min_points_per_voxel: int,
model_name: str,
) -> Tuple[torch.Tensor, torch.Tensor, torch.Tensor]:
M_np = M_matrix.cpu().numpy()
M_inv = np.linalg.inv(M_np)
# clean front points
pc_front, = clean_node.clean_pointcloud(
pc_front,
voxel_size=voxel_size,
min_points_per_voxel=min_points_per_voxel,
width=4096,
height=4096,
)
img, mask, depth_map, pc_front = self._prepare_view_data(
current_enriched_pc, M_np, transform_node, clean_node, proj_node,
voxel_size, min_points_per_voxel, camera_type, horizontal_fov,
width, height
)
# project to image + depth
img, mask, depth_map = proj_node.project_pointcloud(
pc_front,
camera_type,
horizontal_fov,
width,
height,
point_size=3,
return_inverse_depth=False,
)
debug_img = img
# fill nan in depthmap with (-1)
# outpaint missing regions
hole_mask = (mask < 0.5).float()
out_img, out_mask = outpaint_node.outpaint_any(
img,
input_projection = camera_type,
input_horiz_fov = horizontal_fov,
output_projection = camera_type,
output_horiz_fov = horizontal_fov,
output_width = width,
output_height = height,
patch_projection = patch_projection,
patch_horiz_fov = patch_horiz_fov,
patch_res = patch_res,
patch_phi = patch_phi,
patch_theta = patch_theta,
prompt = prompt,
num_inference_steps = num_inference_steps,
cached = False,
guidance_scale = guidance_scale,
mask_blur = mask_blur,
mask = hole_mask,
debug = False,
)
debug_img = out_img
# estimate and renormalize depth
nan_mask = torch.isnan(depth_map)
# …and replace them with –1.0 (in-place)
depth_map[nan_mask] = 0
# clip from -1 to 1000
depth_map = torch.clamp(depth_map, 0, 1000.0)
new_depth, = depth_node.estimate_depth(out_img, model_name, depth_scale=1.0)
new_depth, = zdepth_node.depth_to_ray_depth(
new_depth,
horizontal_fov,
)
# renormalize depth
norm_depth, = renorm_node.renormalize_depth(
new_depth,
depth_map,
depth_mask=(mask>=0.5)*1,
guidance_mask=(mask<0.5)*1,
use_inverse=False,
)
# median blur on depth
k = 5
d = norm_depth.permute(0,3,1,2) # [B,1,H,W]
pad = k//2
pd = F.pad(d, (pad, pad, pad, pad), mode='reflect')
patches = pd.unfold(2, k, 1).unfold(3, k, 1)
patches = patches.contiguous().view(d.shape[0], d.shape[1], d.shape[2], d.shape[3], k*k)
d, _ = patches.median(dim=-1)
norm_depth = d.permute(0,2,3,1) # [B,H,W,1]
debug_depth = norm_depth*hole_mask.unsqueeze(0).unsqueeze(-1)+depth_map*(1-hole_mask.unsqueeze(0).unsqueeze(-1))
# fill nan in depthmap with (-1)
# outpaint missing regions
hole_mask = (mask < 0.5).float()
out_img = self._outpaint_missing_regions(
img, hole_mask, # Pass hole_mask instead of the full mask
outpaint_node, camera_type, horizontal_fov, width, height,
patch_projection, patch_horiz_fov, patch_res,
patch_phi, patch_theta, prompt,
num_inference_steps, guidance_scale, mask_blur
)
# estimate and renormalize depth
norm_depth, debug_depth_view = self._estimate_and_refine_depth(
out_img, depth_map, mask, hole_mask,
depth_node, zdepth_node, renorm_node,
model_name, horizontal_fov
)
# back to pointcloud
pc_new, = depth2pc_node.depth_to_pointcloud(
out_img,
camera_type,
horizontal_fov,
depth_scale=1.0,
invert_depth=False,
depthmap=norm_depth,
mask=hole_mask,
)
# back to pointcloud
pc_world = self._convert_depth_to_world_pointcloud(
out_img, norm_depth, hole_mask, M_inv,
depth2pc_node, transform_node,
camera_type, horizontal_fov
)
# enriched_pc is not rotated
current_enriched_pc = torch.cat([current_enriched_pc, pc_world.to(device)], dim=0)
return current_enriched_pc, out_img, debug_depth_view # Return out_img and the depth for this view
pc_world, = transform_node.transform_pointcloud(pc_new, M_inv)
# enriched_pc is not rotated
enriched_pc = torch.cat([enriched_pc, pc_world.to(device)], dim=0)
return enriched_pc, debug_img, norm_depth
def _convert_depth_to_world_pointcloud(
self,
out_img: torch.Tensor,
norm_depth: torch.Tensor,
hole_mask: torch.Tensor,
M_inv: np.ndarray,
depth2pc_node: DepthToPointCloud,
transform_node: TransformPointCloud,
camera_type: str,
horizontal_fov: float,
) -> torch.Tensor:
pc_new, = depth2pc_node.depth_to_pointcloud(
out_img,
camera_type,
horizontal_fov,
depth_scale=1.0,
invert_depth=False,
depthmap=norm_depth,
mask=hole_mask,
)
pc_world, = transform_node.transform_pointcloud(pc_new, M_inv)
return pc_world
def _estimate_and_refine_depth(
self,
out_img: torch.Tensor,
depth_map: torch.Tensor,
original_mask: torch.Tensor, # Mask from projection
hole_mask: torch.Tensor,
depth_node: DepthEstimatorNode,
zdepth_node: ZDepthToRayDepthNode,
renorm_node: DepthRenormalizer,
model_name: str,
horizontal_fov: float,
) -> Tuple[torch.Tensor, torch.Tensor]:
nan_mask = torch.isnan(depth_map)
depth_map[nan_mask] = 0 # In-place modification
depth_map = torch.clamp(depth_map, 0, 1000.0)
new_depth, = depth_node.estimate_depth(out_img, model_name, depth_scale=1.0)
new_depth, = zdepth_node.depth_to_ray_depth(
new_depth,
horizontal_fov,
)
# renormalize depth
# Use original_mask for depth_mask as it represents valid projected areas
norm_depth, = renorm_node.renormalize_depth(
new_depth,
depth_map,
depth_mask=(original_mask >= 0.5) * 1,
guidance_mask=(hole_mask >= 0.5) * 1, # hole_mask is appropriate here
use_inverse=False,
)
# median blur on depth
k = 5
d = norm_depth.permute(0,3,1,2) # [B,1,H,W]
pad = k//2
pd = F.pad(d, (pad, pad, pad, pad), mode='reflect')
patches = pd.unfold(2, k, 1).unfold(3, k, 1)
patches = patches.contiguous().view(d.shape[0], d.shape[1], d.shape[2], d.shape[3], k*k)
d_median, _ = patches.median(dim=-1) # Renamed to avoid conflict
norm_depth_blurred = d_median.permute(0,2,3,1) # [B,H,W,1]
# Create debug_depth_view using the blurred normalized depth for holes
# and the original depth_map for non-holes.
debug_depth_view = norm_depth_blurred * hole_mask.unsqueeze(0).unsqueeze(-1) + \
depth_map * (1 - hole_mask.unsqueeze(0).unsqueeze(-1))
return norm_depth_blurred, debug_depth_view
def _outpaint_missing_regions(
self,
img: torch.Tensor,
hole_mask: torch.Tensor, # Expects the specific hole_mask
outpaint_node: OutpaintAnyProjection,
camera_type: str,
horizontal_fov: float,
width: int,
height: int,
patch_projection: str,
patch_horiz_fov: float,
patch_res: int,
patch_phi: float,
patch_theta: float,
prompt: str,
num_inference_steps: int,
guidance_scale: float,
mask_blur: int,
) -> torch.Tensor: # Returns only out_img, out_mask is not used later
out_img, _ = outpaint_node.outpaint_any( # Assign out_mask to _
img,
input_projection = camera_type,
input_horiz_fov = horizontal_fov,
output_projection = camera_type,
output_horiz_fov = horizontal_fov,
output_width = width,
output_height = height,
patch_projection = patch_projection,
patch_horiz_fov = patch_horiz_fov,
patch_res = patch_res,
patch_phi = patch_phi,
patch_theta = patch_theta,
prompt = prompt,
num_inference_steps = num_inference_steps,
cached = False,
guidance_scale = guidance_scale,
mask_blur = mask_blur,
mask = hole_mask, # Use the passed hole_mask
debug = False,
)
return out_img
def _prepare_view_data(
self,
current_enriched_pc: torch.Tensor,
M_np: np.ndarray,
transform_node: TransformPointCloud,
clean_node: PointCloudCleaner,
proj_node: ProjectPointCloud,
voxel_size: float,
min_points_per_voxel: int,
camera_type: str,
horizontal_fov: float,
width: int,
height: int,
) -> Tuple[torch.Tensor, torch.Tensor, torch.Tensor, torch.Tensor]:
# transform and select front points
rotated, = transform_node.transform_pointcloud(current_enriched_pc, M_np)
pc_front = rotated[rotated[:, 2] > 0]
# clean front points
pc_front, = clean_node.clean_pointcloud(
pc_front,
voxel_size=voxel_size,
min_points_per_voxel=min_points_per_voxel,
width=4096, # Consider passing these as params if they vary
height=4096, # Consider passing these as params if they vary
)
# project to image + depth
img, mask, depth_map = proj_node.project_pointcloud(
pc_front,
camera_type,
horizontal_fov,
width,
height,
point_size=3,
return_inverse_depth=False,
)
return img, mask, depth_map, pc_front
NODE_CLASS_MAPPINGS = {"FisheyeDepthEstimator": FisheyeDepthEstimator,
"PointcloudTrajectoryEnricher": PointcloudTrajectoryEnricher}
Binary file not shown.
Binary file not shown.

Before

Width:  |  Height:  |  Size: 224 KiB

Binary file not shown.

Before

Width:  |  Height:  |  Size: 13 MiB

+29 -11
View File
@@ -129,7 +129,7 @@ class OutpaintAnyProjection:
patch_mask = torch.ones_like(patch_mask) * 1 if debug else patch_mask
# 5) Reproject inpainted patch back
back_img, back_mask = reproj.reproject_image(
back_img, inpainted_patch_reproj_mask_raw = reproj.reproject_image( # Store raw mask output
inpainted_patch,
patch_horiz_fov, output_horiz_fov,
patch_projection, output_projection,
@@ -138,19 +138,37 @@ class OutpaintAnyProjection:
transform_matrix=rot_m,
feathering=0,
)
back_mask = normalize_mask(back_mask).bool() # True where patch contributes
# inpainted_patch_coverage_mask: 1.0 where the reprojected inpainted patch has content, 0.0 otherwise.
inpainted_patch_coverage_mask = normalize_mask(back_mask) # Ensure it's float [0,1]
# original coverage: True = had data, False = hole
orig_covered = ~base_mask.bool()
base_img=base_img * orig_covered.unsqueeze(-1)
# fill only holes where back_mask is False
filled = back_img * (~back_mask.unsqueeze(-1))*base_mask.unsqueeze(-1)
final_img = base_img+filled
# --- Define Masks based on Conventions ---
# base_mask: 1.0 where original reprojected image has content, 0.0 for holes.
# initial_hole_mask: 1.0 where original reprojected image has holes (inverse of base_mask).
initial_hole_mask = 1.0 - base_mask
# inpainted_patch_coverage_mask: 1.0 where reprojected inpainted patch has content.
# anything that’s still a hole after back‐projection needs inpaint
needs_inpaint = (~((orig_covered) | (~back_mask))).to(torch.float32)
# --- Compositing Logic ---
# Goal: Inpainted patch takes precedence in overlapping areas. Original content is used elsewhere.
return final_img, needs_inpaint
# Contribution from the original image:
# Valid original pixels, excluding areas covered by the inpainted patch.
original_content_contribution = base_img * base_mask.unsqueeze(-1) * \
(1.0 - inpainted_patch_coverage_mask.unsqueeze(-1))
# Contribution from the inpainted patch (reprojected as back_img):
# Valid inpainted pixels, where the patch provides coverage.
inpainted_patch_contribution = back_img * inpainted_patch_coverage_mask.unsqueeze(-1)
# Combine:
final_img = original_content_contribution + inpainted_patch_contribution
# --- needs_inpaint_mask Derivation ---
# Identifies areas that were initially holes AND remain un-filled by the reprojected inpainted patch.
# These are areas that still require inpainting if a further pass was to be made.
not_covered_by_inpainted_patch = 1.0 - inpainted_patch_coverage_mask
needs_inpaint_mask = initial_hole_mask * not_covered_by_inpainted_patch # Element-wise multiplication (AND logic)
return final_img, needs_inpaint_mask
# register
NODE_CLASS_MAPPINGS = {
-125
View File
@@ -1,125 +0,0 @@
#!/usr/bin/env bash
set -euo pipefail
# ----------------------------------------
# Functions
# ----------------------------------------
install_pytorch() {
echo "Installing PyTorch, TorchVision, TorchAudio..."
pip3 install -U torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
pip3 install bitsandbytes
pip3 install accelerate
}
install_system_deps() {
echo "Updating apt and installing system dependencies..."
sudo apt-get update
sudo apt-get install -y build-essential ffmpeg libsm6 libxext6
}
clone_and_install_comfyui() {
echo "Cloning ComfyUI..."
git clone https://github.com/comfyanonymous/ComfyUI.git
echo "Installing ComfyUI requirements..."
pip3 install -r ComfyUI/requirements.txt
}
install_camera_node() {
echo "Cloning camera‑ComfyUI..."
git clone https://github.com/Alexankharin/camera-comfyUI.git \
ComfyUI/custom_nodes/camera-comfyUI
echo "Installing camera‑ComfyUI requirements..."
pip3 install -r ComfyUI/custom_nodes/camera-comfyUI/requirements.txt
}
install_image_filters() {
echo "Cloning Image‑Filters node..."
git clone https://github.com/spacepxl/ComfyUI-Image-Filters.git \
ComfyUI/custom_nodes/ComfyUI-Image-Filters
echo "Installing Image‑Filters requirements..."
pip3 install -r ComfyUI/custom_nodes/ComfyUI-Image-Filters/requirements.txt
}
clone_flux_inpainting() {
echo "Cloning ComfyUI‑Flux‑Inpainting..."
git clone https://github.com/rubi-du/ComfyUI-Flux-Inpainting.git \
ComfyUI/custom_nodes/Flux-Inpainting
echo "Renaming Flux‑Inpainting folder..."
mv ComfyUI/custom_nodes/Flux-Inpainting \
ComfyUI/custom_nodes/inpainting_flux
}
install_hf_hub() {
echo "Installing huggingface_hub..."
pip3 install huggingface_hub
}
download_vae_models() {
echo "Downloading WAN‑VACE models via wget..."
wget -O ComfyUI/models/vae/wan_2.1_vae.safetensors \
"https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/resolve/main/split_files/vae/wan_2.1_vae.safetensors?download=true"
wget -O ComfyUI/models/text_encoders/umt5_xxl_fp16.safetensors \
"https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/resolve/main/split_files/text_encoders/umt5_xxl_fp16.safetensors?download=true"
wget -O ComfyUI/models/diffusion_models/wan2.1_vace_14B_fp16.safetensors \
"https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/resolve/main/split_files/diffusion_models/wan2.1_vace_14B_fp16.safetensors"
}
login_hf_hub() {
echo "Logging in to Hugging Face Hub..."
huggingface-cli login
}
# ----------------------------------------
# Main CLI
# ----------------------------------------
# default to "install" if no arg given
MODE="${1:-install}"
case "$MODE" in
modules)
install_pytorch
install_system_deps
clone_and_install_comfyui
install_camera_node
install_image_filters
install_hf_hub
;;
flux)
clone_flux_inpainting
;;
vae)
download_vae_models
;;
install)
install_pytorch
install_system_deps
clone_and_install_comfyui
install_camera_node
clone_flux_inpainting
install_image_filters
install_hf_hub
;;
all)
install_pytorch
install_system_deps
clone_and_install_comfyui
install_camera_node
clone_flux_inpainting
install_image_filters
install_hf_hub
download_vae_models
login_hf_hub
echo "All done! 🎉"
;;
*)
echo "Usage: $0 {install|modules|flux|vae|all}"
exit 1
;;
esac
+5 -5
View File
@@ -41,7 +41,7 @@ class DepthEstimatorNode:
RETURN_TYPES = ("TENSOR",)
RETURN_NAMES = ("depth tensor",)
FUNCTION = "estimate_depth"
CATEGORY = "Camera/Depth"
CATEGORY = "Camera/depth"
def estimate_depth(
self,
@@ -110,7 +110,7 @@ class DepthToImageNode:
RETURN_TYPES = ("IMAGE",)
RETURN_NAMES = ("depth image",)
FUNCTION = "depth_to_image"
CATEGORY = "Camera/Depth"
CATEGORY = "Camera/depth"
def depth_to_image(
self,
@@ -158,7 +158,7 @@ class ZDepthToRayDepthNode:
RETURN_TYPES = ("TENSOR",)
RETURN_NAMES = ("ray depth",)
FUNCTION = "depth_to_ray_depth"
CATEGORY = "Camera/Depth"
CATEGORY = "Camera/depth"
def depth_to_ray_depth(
self,
@@ -229,7 +229,7 @@ class CombineDepthsNode:
RETURN_TYPES = ("TENSOR","MASK")
RETURN_NAMES = ("combined_depth","combined_mask")
FUNCTION = "combine_depths"
CATEGORY = "Camera/Depth"
CATEGORY = "Camera/depth"
def combine_depths(
self,
@@ -358,7 +358,7 @@ class DepthRenormalizer:
RETURN_TYPES = ("TENSOR",)
RETURN_NAMES = ("depth tensor",)
FUNCTION = "renormalize_depth"
CATEGORY = "Camera/Depth"
CATEGORY = "Camera/depth"
def renormalize_depth(
self,
+56 -126
View File
@@ -113,7 +113,7 @@ def XYZ_to_equirect(X: torch.Tensor, Y: torch.Tensor, Z: torch.Tensor, fov: floa
Convert XYZ coordinates to normalized UV and depth using equirectangular projection.
"""
# full 360°×180°
fov_rad = math.radians(fov) / 2
fov_rad = math.radians(fov)
depth = torch.sqrt(X**2 + Y**2 + Z**2)
lon = torch.atan2(X, Z) # –π → +π
lat = torch.asin(Y / depth) # –π/2 → +π/2
@@ -167,7 +167,7 @@ class DepthToPointCloud:
RETURN_TYPES = ("TENSOR",)
RETURN_NAMES = ("pointcloud",)
FUNCTION = "depth_to_pointcloud"
CATEGORY = "Camera/PointCloud"
CATEGORY = "Camera/pointcloud"
def depth_to_pointcloud(
self,
@@ -273,7 +273,7 @@ class TransformPointCloud:
RETURN_TYPES = ("TENSOR",)
RETURN_NAMES = ("transformed pointcloud",)
FUNCTION = "transform_pointcloud"
CATEGORY = "Camera/PointCloud"
CATEGORY = "Camera/pointcloud"
def transform_pointcloud(
self,
@@ -321,7 +321,7 @@ class ProjectPointCloud:
RETURN_TYPES = ("IMAGE", "MASK", "TENSOR")
RETURN_NAMES = ("image", "mask", "depth")
FUNCTION = "project_pointcloud"
CATEGORY = "Camera/PointCloud"
CATEGORY = "Camera/pointcloud"
def project_pointcloud(
self,
@@ -393,7 +393,6 @@ class ProjectPointCloud:
img4 = flat.view(output_height, output_width, 4)
rgb = img4[..., :3].clamp(0, 255)
alpha = (img4[..., 3] > 0).float()
mask_init= (img4[..., 3] > 0) # initial mask
rgb *= alpha.unsqueeze(-1)
depth_img = z_front.view(output_height, output_width)
rgb_HR = rgb
@@ -407,6 +406,7 @@ class ProjectPointCloud:
idxbuf.fill_(-1)
idxbuf.scatter_reduce_(0, pix, order_m, reduce='amax', include_self=True)
win_back = idxbuf[pix] >= 0
flat.fill_(0)
flat[pix[win_back]] = colors[win_back]
back4 = flat.view(output_height, output_width, 4)
@@ -430,29 +430,8 @@ class ProjectPointCloud:
# merge only at hole locations
rgb[hole] = rgb_med[hole]
# alpha already set to 1.0 for holes
# 8 apply median blur to mask if point_size > 1 and to initial image
mask_t = alpha.unsqueeze(0).unsqueeze(0) # [1,1,H,W]
pad = point_size // 2
ksize = (point_size, point_size)
# b) grow (dilate) mask by max‑pool
mask_grow = F.max_pool2d(mask_t, kernel_size=ksize, stride=1, padding=pad)
# c) shrink (erode) by inverting, max‑pool, then inverting back
mask_shrink = 1.0 - F.max_pool2d(1.0 - mask_grow, kernel_size=ksize, stride=1, padding=pad)
# d) back to [H,W] and use as our new alpha
alpha = mask_shrink.squeeze(0).squeeze(0)
print(1)
# e) median‑filter the *whole* RGB image
# prep for kornia: [B,C,H,W]
rgb_t_full = rgb.permute(2,0,1).unsqueeze(0) # [1,3,H,W]
rgb_med_full = median_blur(rgb_t_full, ksize) # [1,3,H,W]
rgb_med_full = rgb_med_full.squeeze(0).permute(1,2,0) # [H,W,3]
# g) refill *only* the original holes with the median result
rgb[~mask_init] = rgb_med_full[~mask_init]
# 9) Pack and return with original script shapes
# 8) Pack and return with original script shapes
img = rgb.unsqueeze(0) # [1,H,W,3]
mask_out = alpha # [H,W]
depth4 = depth_img.unsqueeze(0).unsqueeze(-1) # [1,H,W,1]
@@ -477,7 +456,7 @@ class PointCloudUnion:
RETURN_TYPES = ("TENSOR",)
RETURN_NAMES =("merged pointcloud",)
FUNCTION = "union_pointclouds"
CATEGORY = "Camera/PointCloud"
CATEGORY = "Camera/pointcloud"
def union_pointclouds(
self,
@@ -513,7 +492,7 @@ class LoadPointCloud:
}
}
CATEGORY = "Camera/PointCloud"
CATEGORY = "Camera/pointcloud"
RETURN_TYPES = ("TENSOR",)
RETURN_NAMES = ("loaded pointcloud",)
FUNCTION = "load_pointcloud"
@@ -525,45 +504,24 @@ class LoadPointCloud:
arr = np.load(file_path)
tensor_pc = torch.from_numpy(arr)
return (tensor_pc,)
if o3d is None:
logging.warning("[camera-comfyUI] open3d is not installed. Falling back to manual PLY parser.")
coords = []
colors = []
with open(file_path, 'r') as f:
coords = []
colors = []
with open(file_path, 'r') as f:
line = f.readline().strip()
while not line.startswith("end_header"):
line = f.readline().strip()
while not line.startswith("end_header"):
line = f.readline().strip()
for line in f:
parts = line.strip().split()
if len(parts) < 7:
continue
x, y, z = map(float, parts[0:3])
r, g, b, a = map(float, parts[3:7])
coords.append((x, y, z))
colors.append((r, g, b, a))
np_coords = np.array(coords, dtype=np.float32)
np_colors = np.array(colors, dtype=np.float32)
# if colors are > 1, normalize them to [0,1]
if np_colors.max() > 1.0:
np_colors = np_colors / 255.0
else:
pc = o3d.t.io.read_point_cloud(file_path)
np_coords = pc.point["positions"].numpy().astype(np.float32)
if "colors" in pc.point:
cols = pc.point["colors"].numpy().astype(np.float32)
else:
cols = np.ones((np_coords.shape[0], 3), dtype=np.float32)
if "alpha" in pc.point:
alpha = pc.point["alpha"].numpy().astype(np.float32)
else:
alpha = np.ones((np_coords.shape[0], 1), dtype=np.float32)
np_colors = np.concatenate([cols, alpha], axis=1)
if np_colors.max() > 1.0:
np_colors = np_colors / 255.0
# combine coords and colors into a single tensor
combined = np.concatenate([np_coords, np_colors], axis=1)
tensor_pc = torch.from_numpy(combined)
for line in f:
parts = line.strip().split()
if len(parts) < 7:
continue
x, y, z = map(float, parts[0:3])
r, g, b, a = map(int, parts[3:7])
coords.append((x, y, z))
colors.append((r, g, b, a))
np_coords = np.array(coords, dtype=np.float32)
np_colors = np.array(colors, dtype=np.float32)/255.0
combined = np.concatenate([np_coords, np_colors], axis=1)
tensor_pc = torch.from_numpy(combined)
return (tensor_pc,)
@classmethod
@@ -611,7 +569,7 @@ class SavePointCloud:
RETURN_TYPES = ()
FUNCTION = "save_pointcloud"
OUTPUT_NODE = True
CATEGORY = "Camera/PointCloud"
CATEGORY = "Camera/pointcloud"
DESCRIPTION = "Saves the input point cloud to your ComfyUI output directory as .ply or .npy."
def save_pointcloud(self, pointcloud: torch.Tensor, filename_prefix: str, save_as: str = "ply"):
@@ -628,36 +586,24 @@ class SavePointCloud:
os.makedirs(full_output_folder, exist_ok=True)
base_name = filename.replace("%batch_num%", "0")
if save_as == "ply":
ply_name = f"{base_name}_{counter:05}.ply"
ply_path = os.path.join(full_output_folder, ply_name)
coords = pointcloud[:, :3].cpu().numpy().astype(np.float32)
colors = pointcloud[:, 3:].cpu().numpy().clip(0, 1).astype(np.float32)
if o3d is None:
logging.warning("[camera-comfyUI] open3d is not installed. Falling back to manual ASCII PLY writer.")
with open(ply_path, 'w') as f:
f.write("ply\n")
f.write("format ascii 1.0\n")
f.write(f"element vertex {coords.shape[0]}\n")
f.write("property float x\n")
f.write("property float y\n")
f.write("property float z\n")
f.write("property float red\n")
f.write("property float green\n")
f.write("property float blue\n")
f.write("property float alpha\n")
f.write("end_header\n")
for (x, y, z), (r, g, b, a) in zip(coords, colors):
f.write(f"{x} {y} {z} {r} {g} {b} {a}\n")
else:
pc = o3d.t.geometry.PointCloud()
pc.point["positions"] = o3d.core.Tensor(coords, o3d.core.float32)
pc.point["colors"] = o3d.core.Tensor(colors[:, :3], o3d.core.float32)
if colors.shape[1] > 3:
pc.point["alpha"] = o3d.core.Tensor(colors[:, 3:], o3d.core.float32)
else:
pc.point["alpha"] = o3d.core.Tensor(np.ones((coords.shape[0], 1), dtype=np.float32), o3d.core.float32)
o3d.t.io.write_point_cloud(ply_path, pc)
ply_name = f"{base_name}_{counter:05}.ply"
ply_path = os.path.join(full_output_folder, ply_name)
coords = pointcloud[:, :3].cpu().numpy()
colors = pointcloud[:, 3:].cpu().numpy().clip(0,1)
with open(ply_path, 'w') as f:
f.write("ply\n")
f.write("format ascii 1.0\n")
f.write(f"element vertex {coords.shape[0]}\n")
f.write("property float x\n")
f.write("property float y\n")
f.write("property float z\n")
f.write("property uchar red\n")
f.write("property uchar green\n")
f.write("property uchar blue\n")
f.write("property uchar alpha\n")
f.write("end_header\n")
for (x,y,z), (r,g,b,a) in zip(coords, colors):
f.write(f"{x} {y} {z} {int(r*255)} {int(g*255)} {int(b*255)} {int(a*255)}\n")
file_name = ply_name
else:
npy_name = f"{base_name}_{counter:05}.npy"
@@ -694,15 +640,12 @@ class CameraMotionNode:
"output_width": ("INT", {"default":512, "min":8, "max":16384}),
"output_height": ("INT", {"default":512, "min":8, "max":16384}),
"point_size": ("INT", {"default":1, "min":1}),
"widen_mask": ("INT", {"default":0, "min":0, "max":64}),
"invert_mask": ("BOOLEAN", {"default": False}),
"points_to_mask": ("BOOLEAN", {"default": False, "tooltip": "Output mask frames of projected points"}),
}}
RETURN_TYPES = ("IMAGE", "MASK")
RETURN_NAMES = ("motion_frames", "mask_frames")
RETURN_TYPES = ("IMAGE",)
RETURN_NAMES = ("motion_frames",)
FUNCTION = "generate_motion_frames"
CATEGORY = "Camera/Trajectory"
CATEGORY = "Camera/pointcloud"
def generate_motion_frames(
self,
@@ -713,10 +656,7 @@ class CameraMotionNode:
output_horizontal_fov: float,
output_width: int,
output_height: int,
point_size: int = 1,
widen_mask: int = 0,
invert_mask: bool = False,
points_to_mask: bool = False
point_size: int = 1
) -> Tuple[torch.Tensor]:
# validate trajectory shape
if trajectory.dim() != 3 or trajectory.shape[1:] != (4,4):
@@ -743,10 +683,9 @@ class CameraMotionNode:
proj_node = ProjectPointCloud()
transform_node = TransformPointCloud()
frames = []
masks = []
for M in tqdm(full_traj):
pc_t, = transform_node.transform_pointcloud(pointcloud, M)
img, mask, _ = proj_node.project_pointcloud(
img, _, _ = proj_node.project_pointcloud(
pc_t,
output_projection,
output_horizontal_fov,
@@ -754,19 +693,10 @@ class CameraMotionNode:
output_height,
point_size
)
if widen_mask > 0:
k = 2 * widen_mask + 1
pad = widen_mask
mask = F.max_pool2d(mask.float().unsqueeze(0).unsqueeze(0), kernel_size=k, stride=1, padding=pad).squeeze(0).squeeze(0)
if invert_mask:
mask = 1.0 - mask
masks.append(mask)
if points_to_mask:
img = mask.unsqueeze(-1).repeat(1,1,1,3)
frames.append(img[0])
# output as (T,H,W,3)
return (torch.stack(frames, dim=0), torch.stack(masks, dim=0))
return (torch.stack(frames, dim=0),)
class CameraInterpolationNode:
"""
@@ -785,7 +715,7 @@ class CameraInterpolationNode:
RETURN_TYPES = ("TENSOR",)
RETURN_NAMES = ("trajectory",)
FUNCTION = "interpolate"
CATEGORY = "Camera/Trajectory"
CATEGORY = "Camera/pointcloud"
def interpolate(
self,
@@ -820,7 +750,7 @@ class CameraTrajectoryNode:
RETURN_TYPES = ("TENSOR",)
RETURN_NAMES = ("trajectory",)
FUNCTION = "build_trajectory"
CATEGORY = "Camera/Trajectory"
CATEGORY = "Camera/pointcloud"
def build_trajectory(
self,
@@ -960,7 +890,7 @@ class PointCloudCleaner:
RETURN_TYPES = ("TENSOR",)
RETURN_NAMES = ("cleaned_pointcloud",)
FUNCTION = "clean_pointcloud"
CATEGORY = "Camera/PointCloud"
CATEGORY = "Camera/pointcloud"
def clean_pointcloud(
self,
@@ -1036,7 +966,7 @@ class ProjectAndClean:
RETURN_TYPES = ("TENSOR",)
RETURN_NAMES = ("cleaned_pointcloud",)
FUNCTION = "project_and_clean"
CATEGORY = "Camera/PointCloud"
CATEGORY = "Camera/pointcloud"
def project_and_clean(
self,
@@ -1150,7 +1080,7 @@ class SaveTrajectory:
RETURN_TYPES = ()
FUNCTION = "save_trajectory"
OUTPUT_NODE = True
CATEGORY = "Camera/Trajectory"
CATEGORY = "Camera/pointcloud"
DESCRIPTION = "Saves the input trajectory tensor (N,4,4) to your ComfyUI output directory as .npy."
def save_trajectory(self, trajectory: torch.Tensor, filename_prefix: str):
@@ -1200,7 +1130,7 @@ class LoadTrajectory:
}
}
CATEGORY = "Camera/Trajectory"
CATEGORY = "Camera/pointcloud"
RETURN_TYPES = ("TENSOR",)
RETURN_NAMES = ("loaded_trajectory",)
FUNCTION = "load_trajectory"
+4 -4
View File
@@ -148,7 +148,7 @@ class ReprojectImage:
RETURN_TYPES: Tuple[str, str] = ("IMAGE", "MASK")
RETURN_NAMES = ("reprojected image", "reprojected mask")
FUNCTION: str = "reproject_image"
CATEGORY: str = "Camera/Reprojection"
CATEGORY: str = "Camera/reproject"
def reproject_image(
self,
@@ -303,7 +303,7 @@ class TransformToMatrix:
RETURN_TYPES: Tuple[str] = ("MAT_4X4",)
RETURN_NAMES = ("transformation matrix",)
FUNCTION: str = "generate_matrix"
CATEGORY: str = "Camera/Matrix"
CATEGORY: str = "Camera/reproject"
def generate_matrix(
self,
@@ -392,7 +392,7 @@ class TransformToMatrixManual:
RETURN_TYPES: Tuple[str] = ("MAT_4X4",)
RETURN_NAMES = ("transformation matrix",)
FUNCTION: str = "generate_matrix"
CATEGORY: str = "Camera/Matrix"
CATEGORY: str = "Camera/reproject"
def generate_matrix(
self,
@@ -444,7 +444,7 @@ class ReprojectDepth:
RETURN_TYPES: Tuple[str, str] = ("TENSOR", "MASK")
RETURN_NAMES = ("reprojected_depth", "reprojected_mask")
FUNCTION: str = "reproject_depth"
CATEGORY: str = "Camera/Reprojection"
CATEGORY: str = "Camera/reproject"
def reproject_depth(
self,
-265
View File
@@ -1,265 +0,0 @@
import torch
import torch.nn.functional as F
import numpy as np
import os
import sys
from typing import Dict, Any, Tuple
from tqdm import tqdm # Added tqdm import
# Import existing pointcloud nodes and projection definitions
from .pointcloud_nodes import DepthToPointCloud, TransformPointCloud, ProjectPointCloud, Projection, PointCloudCleaner
import folder_paths
# Ensure video_depth_anything is on path
video_depth_path = os.path.join("/root/Video-Depth-Anything", "metric_depth")
if video_depth_path not in sys.path:
sys.path.append(video_depth_path)
try:
from video_depth_anything.video_depth import VideoDepthAnything
print("video_depth_anything module loaded successfully.")
except ImportError:
VideoDepthAnything = None
print("Warning: video_depth_anything module not found. Ensure it is installed correctly.")
print("error: ", sys.exc_info()[1])
class VideoCameraMotionSequence:
"""
Takes a sequence of RGB frames and corresponding depth maps,
converts each frame+depth to a pointcloud, interpolates a camera
trajectory to match video length, cleans the pointcloud if needed,
and outputs reprojected images, masks, and depth maps per frame.
"""
@classmethod
def INPUT_TYPES(cls) -> Dict[str, Any]:
return {
"required": {
# Sequence of frames: Tensor [T, H, W, 3]
"frames": ("IMAGE", {"shape_hint": [None, None, None, 3]}),
# Sequence of depth maps: Tensor [T, H, W] or [T, H, W, 1]
"depth_seq": ("TENSOR", {"shape_hint": [None, None, None]}),
# Camera trajectory waypoints: Tensor [K, 4, 4]
"trajectory": ("TENSOR", {"shape_hint": [None, 4, 4]}),
# Input projection parameters
"input_projection": (Projection.PROJECTIONS, {}),
"input_horizontal_fov": ("FLOAT", {"default": 90.0}),
"depth_scale": ("FLOAT", {"default": 1.0}),
"invert_depth": ("BOOLEAN", {"default": False}),
# Output projection parameters
"output_projection": (Projection.PROJECTIONS, {}),
"output_horizontal_fov": ("FLOAT", {"default": 90.0}),
"output_width": ("INT", {"default": 512, "min": 1}),
"output_height": ("INT", {"default": 512, "min": 1}),
"point_size": ("INT", {"default": 1, "min": 1}),
# Cleaning parameters
"voxel_size": ("FLOAT", {"default": 1.0, "min": 1e-3}),
"min_points_per_voxel": ("INT", {"default": 3, "min": 1}),
}
}
RETURN_TYPES = ("IMAGE", "MASK", "TENSOR")
RETURN_NAMES = ("video_frames", "mask_frames", "depths")
FUNCTION = "process_sequence"
CATEGORY = "Camera/Video"
def process_sequence(
self,
frames: torch.Tensor,
depth_seq: torch.Tensor,
trajectory: torch.Tensor,
input_projection: str,
input_horizontal_fov: float,
depth_scale: float,
invert_depth: bool,
output_projection: str,
output_horizontal_fov: float,
output_width: int,
output_height: int,
point_size: int,
voxel_size: float,
min_points_per_voxel: int,
) -> Tuple[torch.Tensor, torch.Tensor, torch.Tensor]:
# frames: [T, H, W, 3]
# depth_seq: [T, H, W] or [T, H, W, 1]
T, H, W, _ = frames.shape
# Interpolate trajectory to match T
K = trajectory.shape[0]
if K < 2:
interp_traj = trajectory.expand(T, 4, 4).clone()
else:
idxs = torch.linspace(0, K - 1, T, device=trajectory.device)
lower = idxs.floor().long().clamp(max=K - 2)
upper = lower + 1
alpha = (idxs - lower.float()).unsqueeze(-1).unsqueeze(-1)
traj_lower = trajectory[lower]
traj_upper = trajectory[upper]
interp_traj = traj_lower * (1 - alpha) + traj_upper * alpha
out_frames = []
out_masks = []
out_depths = []
# Add tqdm progress bar for the sequence
for frame, depth, pose in tqdm(zip(frames, depth_seq, interp_traj), total=T, desc="Processing video frames"):
if depth.dim() == 3 and depth.shape[-1] == 1:
depth = depth.squeeze(-1)
# to pointcloud
pc, = DepthToPointCloud().depth_to_pointcloud(
image=frame.permute(2, 0, 1),
input_projection=input_projection,
input_horizontal_fov=input_horizontal_fov,
depth_scale=depth_scale,
invert_depth=invert_depth,
depthmap=depth,
mask=None,
)
# optional cleaning
if min_points_per_voxel > 1:
pc, = PointCloudCleaner().clean_pointcloud(
pointcloud=pc,
width=output_width,
height=output_height,
voxel_size=voxel_size,
min_points_per_voxel=min_points_per_voxel,
)
# transform and project
pc_t, = TransformPointCloud().transform_pointcloud(pc, pose)
img_t, mask_t, depth_t = ProjectPointCloud().project_pointcloud(
pointcloud=pc_t,
output_projection=output_projection,
output_horizontal_fov=output_horizontal_fov,
output_width=output_width,
output_height=output_height,
point_size=point_size,
)
out_frames.append(img_t[0])
out_masks.append(mask_t)
out_depths.append(depth_t)
return (
torch.stack(out_frames, dim=0), # [T, 3, H, W]
torch.stack(out_masks, dim=0), # [T, H, W]
torch.stack(out_depths, dim=0), # [T, H, W]
)
class DepthFramesToVideo:
"""
Converts a sequence of depth maps into video frame tensors for saving.
"""
@classmethod
def INPUT_TYPES(cls) -> Dict[str, Any]:
return {
"required": {
"depth_seq": ("TENSOR", {"shape_hint": [None, None, None]}),
"mask_seq": ("MASK", {"shape_hint": [None, None, None]}),
"normalize": ("BOOLEAN", {"default": True}),
"invert_depth": ("BOOLEAN", {"default": False}),
}
}
RETURN_TYPES = ("TENSOR", "IMAGE")
RETURN_NAMES = ("video_frames", "depth_video")
FUNCTION = "depth_to_video_frames"
CATEGORY = "Camera/Video"
def depth_to_video_frames(
self,
depth_seq: torch.Tensor,
normalize: bool,
invert_depth: bool,
mask_seq: torch.Tensor,
) -> Tuple[torch.Tensor, torch.Tensor]:
ds = depth_seq.clone().squeeze()
if ds.dim() == 2:
ds = ds.unsqueeze(0) # [H, W] -> [1, H, W]
if ds.dim() != 3:
raise ValueError(f"Expected ds to be 3D [T, H, W], got shape {ds.shape}")
if invert_depth:
ds= 1.0 / (ds + 1e-8) # Avoid division by zero
if normalize:
# Mask: only normalize where depth > 0
mask = mask_seq>0.5
if mask.any():
#percentile first 10 percent min
# sample
minv = ds[mask]
# sample 10000 and find 10% quantile
if minv.numel() > 10000:
minv = minv[torch.randperm(minv.numel())[:10000]]
minv = minv.quantile(0.2)
minv = minv if minv > 0.1 else 0.1 # Avoid division by zero
#percentile last 10 percent max
maxv = ds[mask]
if maxv.numel() > 10000:
maxv = maxv[torch.randperm(maxv.numel())[:10000]]
maxv = maxv.quantile(0.98)
maxv = maxv if maxv < 100 else 100
print(f"Normalizing depth: min={minv}, max={maxv}")
ds_norm = (ds - minv) / (maxv - minv + 1e-8)
ds = ds_norm.clamp(0, 1) # torch.where(mask, ds_norm, ds) # Only normalize valid values
else:
print("Warning: No valid depth values for normalization.")
# expand to 3 channels: [T, H, W] -> [T, 3, H, W]
raw = depth_seq.clone().squeeze()
ds_u8 = (ds * 255.0).round().to(torch.uint8)
raw_u8 = (raw.clamp(0, 255)).to(torch.uint8) # if raw is already in a displayable range
# expand to 3 channels and permute to HWC
ds_color = ds_u8.unsqueeze(1).repeat(1, 3, 1, 1).permute(0, 2, 3, 1)
raw_color = raw_u8.unsqueeze(1).repeat(1, 3, 1, 1).permute(0, 2, 3, 1)
return raw_color, ds_color # [T, 3, H, W] -> [T, H, W, 3]
class VideoMetricDepthEstimate:
"""
Estimates metric depth for a sequence of frames using VideoDepthAnything.
"""
@classmethod
def INPUT_TYPES(cls) -> Dict[str, Any]:
# model files (.pth) in input directory
model_dir = os.path.join(os.getcwd(), "models", "checkpoints")
os.makedirs(model_dir, exist_ok=True)
files = [f for f in os.listdir(model_dir) if f.lower().endswith(('.pth', '.ckpt', '.safetensors'))]
return {
"required": {
"frames": ("IMAGE", {"shape_hint": [None, None, None, 3]}),
"model_checkpoint": (files, {"file_chooser": True}),
"input_size": ("INT", {"default": 518, "min": 64, "max": 2048}),
"max_fps": ("INT", {"default": 60, "min": 1}),
}
}
RETURN_TYPES = ("TENSOR", "FLOAT")
RETURN_NAMES = ("metric_depths", "fps")
FUNCTION = "estimate_metric_depth"
CATEGORY = "Camera/Video"
def estimate_metric_depth(
self,
frames: torch.Tensor,
model_checkpoint: str,
input_size: int,
max_fps: int,
) -> Tuple[torch.Tensor, float]:
if VideoDepthAnything is None:
raise ImportError("VideoDepthAnything library not found")
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
# if max input<1.5 normalize to 0-255
if frames.max() < 1.5:
frames = (frames * 255)
model = VideoDepthAnything(**{"encoder": "vitl", "features": 256, "out_channels": [256,512,1024,1024]})
state = torch.load("/root/ComfyUI/models/checkpoints/{}".format(model_checkpoint), map_location='cpu')
model.load_state_dict(state, strict=True)
model = model.to(device).eval()
np_frames = frames.cpu().numpy().astype(np.uint8)
metric_depths, fps = model.infer_video_depth(np_frames, max_fps, input_size=input_size, device=device.type, fp32=False)
return (torch.from_numpy(metric_depths), float(fps))
# Register nodes
NODE_CLASS_MAPPINGS = {
"VideoCameraMotionSequence": VideoCameraMotionSequence,
"VideoMetricDepthEstimate": VideoMetricDepthEstimate,
"DepthFramesToVideo": DepthFramesToVideo,
}
+1 -289
View File
@@ -1,289 +1 @@
{
"id": "dd56c0bf-7405-406e-924f-42b2feacb73f",
"revision": 0,
"last_node_id": 8,
"last_link_id": 9,
"nodes": [
{
"id": 3,
"type": "SaveWEBM",
"pos": [
606.5880737304688,
1515.5966796875
],
"size": [
315,
437
],
"flags": {},
"order": 5,
"mode": 0,
"inputs": [
{
"name": "images",
"type": "IMAGE",
"link": 5
}
],
"outputs": [],
"properties": {},
"widgets_values": [
"ComfyUI",
"vp9",
10.000000000000002,
32
]
},
{
"id": 7,
"type": "CameraMotionNode",
"pos": [
176.07933044433594,
1504.22705078125
],
"size": [
278.75,
270
],
"flags": {},
"order": 4,
"mode": 0,
"inputs": [
{
"name": "pointcloud",
"type": "TENSOR",
"link": 6
},
{
"name": "trajectory",
"type": "TENSOR",
"link": 9
}
],
"outputs": [
{
"name": "motion_frames",
"type": "IMAGE",
"links": [
5
]
},
{
"name": "mask_frames",
"type": "MASK",
"links": null
}
],
"properties": {
"Node name for S&R": "CameraMotionNode"
},
"widgets_values": [
10,
"PINHOLE",
90,
512,
512,
1,
0,
false,
false
]
},
{
"id": 6,
"type": "LoadPointCloud",
"pos": [
-357.24761962890625,
1294.427734375
],
"size": [
315,
58
],
"flags": {},
"order": 0,
"mode": 0,
"inputs": [],
"outputs": [
{
"name": "loaded pointcloud",
"type": "TENSOR",
"links": [
6
]
}
],
"properties": {
"Node name for S&R": "LoadPointCloud"
},
"widgets_values": [
"ComfyUIPointCloud_00001.ply"
]
},
{
"id": 1,
"type": "TransformToMatrix",
"pos": [
-537.2319946289062,
1491.40380859375
],
"size": [
315,
154
],
"flags": {},
"order": 1,
"mode": 0,
"inputs": [],
"outputs": [
{
"name": "transformation matrix",
"type": "MAT_4X4",
"links": [
7
]
}
],
"properties": {
"Node name for S&R": "TransformToMatrix"
},
"widgets_values": [
0,
0,
0,
0,
0
]
},
{
"id": 5,
"type": "TransformToMatrix",
"pos": [
-531.216796875,
1701.1632080078125
],
"size": [
315,
154
],
"flags": {},
"order": 2,
"mode": 0,
"inputs": [],
"outputs": [
{
"name": "transformation matrix",
"type": "MAT_4X4",
"links": [
8
]
}
],
"properties": {
"Node name for S&R": "TransformToMatrix"
},
"widgets_values": [
0.10000000000000002,
0,
0,
0,
0
]
},
{
"id": 8,
"type": "CameraInterpolationNode",
"pos": [
-121.91971588134766,
1597.3514404296875
],
"size": [
200.21640014648438,
46
],
"flags": {},
"order": 3,
"mode": 0,
"inputs": [
{
"name": "initial_matrix",
"type": "MAT_4X4",
"link": 7
},
{
"name": "final_matrix",
"type": "MAT_4X4",
"link": 8
}
],
"outputs": [
{
"name": "trajectory",
"type": "TENSOR",
"links": [
9
]
}
],
"properties": {
"Node name for S&R": "CameraInterpolationNode"
},
"widgets_values": []
}
],
"links": [
[
5,
7,
0,
3,
0,
"IMAGE"
],
[
6,
6,
0,
7,
0,
"TENSOR"
],
[
7,
1,
0,
8,
0,
"MAT_4X4"
],
[
8,
5,
0,
8,
1,
"MAT_4X4"
],
[
9,
8,
0,
7,
1,
"TENSOR"
]
],
"groups": [],
"config": {},
"extra": {
"ds": {
"scale": 1.015255979947716,
"offset": [
636.7531305750655,
-1186.3099424359816
]
},
"frontendVersion": "1.21.7"
},
"version": 0.4
}
{"id":"dd56c0bf-7405-406e-924f-42b2feacb73f","revision":0,"last_node_id":6,"last_link_id":4,"nodes":[{"id":1,"type":"TransformToMatrix","pos":[-337.9580993652344,1500.5211181640625],"size":[315,154],"flags":{},"order":0,"mode":0,"inputs":[],"outputs":[{"localized_name":"MAT_4X4","name":"MAT_4X4","type":"MAT_4X4","links":[1]}],"properties":{"Node name for S&R":"TransformToMatrix"},"widgets_values":[0,0,0,0,0]},{"id":2,"type":"CameraMotion","pos":[130.0218505859375,1516.7567138671875],"size":[367.79998779296875,218],"flags":{},"order":3,"mode":0,"inputs":[{"localized_name":"pointcloud","name":"pointcloud","type":"TENSOR","link":4},{"localized_name":"initial_matrix","name":"initial_matrix","type":"MAT_4X4","link":1},{"localized_name":"final_matrix","name":"final_matrix","type":"MAT_4X4","link":2}],"outputs":[{"localized_name":"IMAGE","name":"IMAGE","type":"IMAGE","links":[3]}],"properties":{"Node name for S&R":"CameraMotion"},"widgets_values":[24,"PINHOLE",90,1024,1024,2]},{"id":3,"type":"SaveWEBM","pos":[606.5880737304688,1515.5966796875],"size":[315,437],"flags":{},"order":4,"mode":0,"inputs":[{"localized_name":"images","name":"images","type":"IMAGE","link":3}],"outputs":[],"properties":{},"widgets_values":["ComfyUI","vp9",10.000000000000002,32]},{"id":5,"type":"TransformToMatrix","pos":[-262.9134521484375,1740.8876953125],"size":[315,154],"flags":{},"order":1,"mode":0,"inputs":[],"outputs":[{"localized_name":"MAT_4X4","name":"MAT_4X4","type":"MAT_4X4","links":[2]}],"properties":{"Node name for S&R":"TransformToMatrix"},"widgets_values":[0.10000000000000002,0,0,0,0]},{"id":6,"type":"LoadPointCloud","pos":[-357.24761962890625,1294.427734375],"size":[315,58],"flags":{},"order":2,"mode":0,"inputs":[],"outputs":[{"localized_name":"TENSOR","name":"TENSOR","type":"TENSOR","links":[4]}],"properties":{"Node name for S&R":"LoadPointCloud"},"widgets_values":["ComfyUIPointCloud_00001.ply"]}],"links":[[1,1,0,2,1,"MAT_4X4"],[2,5,0,2,2,"MAT_4X4"],[3,2,0,3,0,"IMAGE"],[4,6,0,2,0,"TENSOR"]],"groups":[],"config":{},"extra":{"ds":{"scale":1.351305709310409,"offset":[-187.6257577580669,-1567.2143321744395]}},"version":0.4}
File diff suppressed because it is too large Load Diff
File diff suppressed because it is too large Load Diff