Gero Doll
ce0b34cbfe
docs: add node preview and logo images to README - Add depth estimation logo in header - Include node preview image - Fix image alignments and sizing - Update markdown formatting
82 lines
2.3 KiB
Markdown
82 lines
2.3 KiB
Markdown
## <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:
|
|
```bash
|
|
git clone https://github.com/yourusername/ComfyUIDepthEstimation.git
|
|
```
|
|
|
|
2. Navigate to the repository directory:
|
|
|
|
```bash
|
|
cd ComfyUIDepthEstimation
|
|
```
|
|
|
|
3. Install the required dependencies:
|
|
|
|
```bash
|
|
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.
|
|
|
|
---
|