Sign in

Nodes Browser

ComfyDeploy: How ComfyUIDepthEstimation works in ComfyUI?

What is ComfyUIDepthEstimation?

A custom depth estimation node for ComfyUI using transformer models. It integrates depth estimation with automatic gamma correction, contrast adjustment, and edge detection, based on the [a/TransformDepth](https://github.com/Limbicnation/TransformDepth) repository.

How to install it in ComfyDeploy?

Head over to the machine page

  1. Click on the "Create a new machine" button
  2. Select the Edit build steps
  3. Add a new step -> Custom Node
  4. Search for ComfyUIDepthEstimation and select it
  5. Close the build step dialig and then click on the "Save" button to rebuild the machine

<img src="images/depth-estimation-icon.svg" width="32" height="32" alt="Depth Estimation Icon" style="vertical-align: middle"> ComfyUIDepthEstimation ComfyUIDepthEstimation

<div align="center"> <img src="images/depth-estimation-icon.svg" width="150" height="150" alt="Depth Estimation Logo"> </div>

A custom depth estimation node for ComfyUI using Depth-Anything models to generate depth maps from images.

Features

  • Multiple model options:
    • Depth-Anything-Small
    • Depth-Anything-Base
    • Depth-Anything-Large
    • Depth-Anything-V2-Small
    • Depth-Anything-V2-Base
  • Post-processing options:
    • Gaussian blur (adjustable radius)
    • Median filtering (configurable size)
    • Automatic contrast enhancement
    • Gamma correction

Installation

  1. Clone the repository:

    git clone https://github.com/yourusername/ComfyUIDepthEstimation.git

  2. Navigate to the repository directory:

    cd ComfyUIDepthEstimation

  3. Install the required dependencies:

    pip install -r requirements.txt

Usage

<div align="center"> <img src="images/depth-estimation-node.png" width="600" alt="Depth Estimation Node Preview"> </div>

Node Parameters

Node Parameters

image: Input image (IMAGE type) model_name: Select from available Depth-Anything models blur_radius: Gaussian blur radius (0.0 - 10.0, default: 2.0) median_size: Median filter size (3, 5, 7, 9, 11) apply_auto_contrast: Enable automatic contrast enhancement apply_gamma: Enable gamma correction

Integrate with ComfyUI

  1. Copy the depth_estimation_node.py file to your ComfyUI custom nodes directory.

  2. Update ComfyUI configuration to include the custom node if necessary.

  3. Restart ComfyUI to load the new custom node.

Example Usage

  1. Add the Depth Estimation node to your ComfyUI workflow
  2. Connect an image source to the node's image input
  3. Configure the parameters:
    • Select a model (e.g., "Depth-Anything-V2-Small")
    • Adjust blur_radius (0-10) for depth map smoothing
    • Choose median_size (3-11) for noise reduction
    • Toggle auto_contrast and gamma correction as needed
  4. Connect the output to a Preview Image node or other image processing nodes

The node will process the input image and output a depth map that can be used in your ComfyUI workflow.

License

This project is licensed under the Apache License.