Skip to content

spacewalk01/depth-pro-webui

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

29 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Web UI for Depth-Pro

Web UI for Depth-Pro Depth Estimation

Metric depth estimation determines real-world distances to objects in a scene from images. This repo provides a web UI that allows users to estimate metric depth and visualize depth maps by easily uploading images using the Depth Pro model through a simple Gradio UI interface.

🛠️ Getting Started

📦 Installation

  1. Create and activate a virtual environment:

    conda create -n depth-pro -y python=3.9
    conda activate depth-pro
  2. Clone this repo and install the required library as follows:

    git clone https://github.com/spacewalk01/depth-pro-webui.git
    cd depth-pro-webui
    pip install -r requirements.txt
  3. Download the model:

    Download the model checkpoint depth_pro.pt and place it in the checkpoints folder.

🚀 Running the Application

  1. Launch the Gradio interface:

    python main.py
  2. Open the provided local URL in your web browser to access the interface.

🖼️ Usage

  1. Upload an Image: Use the image uploader to select an image for depth estimation.
  2. Adjust Options:
    • Auto Rotate: Enable or disable auto-rotation of the image.
    • Remove Alpha: Enable or disable the removal of the alpha channel.
  3. View Results:
    • The depth map will be displayed on the interface.
    • The focal length in pixels will also be shown.

📜 License

This project is licensed under the Apple License. See the LICENSE file for details.

🙏 Acknowledgments

  • Gradio for providing an easy interface for machine learning models.
  • Depth Pro for enabling sharp monocular metric depth estimation in less than a second.

Releases

No releases published

Packages

No packages published

Languages