Spaces:
Runtime error
Runtime error
| import gradio as gr | |
| from transformers import pipeline | |
| import torch | |
| import numpy as np | |
| from PIL import Image | |
| import gradio as gr | |
| from gradio_client import Client | |
| import os | |
| import spaces | |
| import json | |
| from gradio_depth_pred import create_demo as create_depth_pred_demo | |
| from gradio_im_to_3d import create_demo as create_im_to_3d_demo | |
| model = torch.hub.load('isl-org/ZoeDepth', "ZoeD_N", pretrained=True).to('cuda').eval() | |
| #dpt_beit = pipeline(task = "depth-estimation", model="Intel/dpt-beit-base-384", device=0) | |
| dpt_beit = pipeline(task = "depth-estimation", model="Intel/dpt-beit-large-512", device=0) | |
| #depth_anything = pipeline(task = "depth-estimation", model="nielsr/depth-anything-small", device=0) | |
| depth_anything = pipeline(task = "depth-estimation", model="LiheYoung/depth-anything-large-hf", device=0) | |
| dpt_large = pipeline(task = "depth-estimation", model="intel/dpt-large", device=0) | |
| def depth_anything_inference(img): | |
| return depth_anything(img)["depth"] | |
| def dpt_beit_inference(img): | |
| return dpt_beit(img)["depth"] | |
| def dpt_large_inference(img): | |
| return dpt_large(img)["depth"] | |
| def infer(img): | |
| if img is None: | |
| return None, None, None | |
| else: | |
| return dpt_large_inference(img), dpt_beit_inference(img), depth_anything_inference(img) | |
| css = """ | |
| #mkd { | |
| height: 500px; | |
| overflow: auto; | |
| border: 1px solid #ccc; | |
| #img-display-container { | |
| max-height: 50vh; | |
| } | |
| #img-display-input { | |
| max-height: 40vh; | |
| } | |
| #img-display-output { | |
| max-height: 40vh; | |
| } | |
| } | |
| """ | |
| css_zoe = """ | |
| #img-display-container { | |
| max-height: 50vh; | |
| } | |
| #img-display-input { | |
| max-height: 40vh; | |
| } | |
| #img-display-output { | |
| max-height: 40vh; | |
| } | |
| """ | |
| with gr.Blocks(css=css) as demo: | |
| gr.HTML("<h1><center>Compare Depth Estimation Models<center><h1>") | |
| gr.Markdown("In this Space, you can compare different depth estimation models: [DPT-Large](https://huggingface.co/Intel/dpt-large), [DPT with BeiT backbone](https://huggingface.co/Intel/dpt-beit-large-512) and the recent [Depth Anything Model small checkpoint](https://huggingface.co/LiheYoung/depth-anything-small-hf). 🤩") | |
| gr.Markdown("You can also see how they compare in terms of speed [here](https://huggingface2.notion.site/DPT-Benchmarks-1e516b0ba193460e865c47b3a5681efb?pvs=4).") | |
| gr.Markdown("Simply upload an image or try one of the examples to see the outputs.") | |
| with gr.Column(): | |
| with gr.Row(): | |
| input_img = gr.Image(label="Input Image", type="pil") | |
| with gr.Row(): | |
| output_1 = gr.Image(type="pil", label="Intel dpt-large") | |
| output_2 = gr.Image(type="pil", label="DPT with BeiT Backbone, dpt-beit-large-512") | |
| output_3 = gr.Image(type="pil", label="LiheYoung/depth-anything-large-hf") | |
| gr.Examples([["bee.jpg"], ["cat.png"], ["cats.png"]], | |
| inputs = input_img, | |
| outputs = [output_1, output_2, output_3], | |
| fn=infer, | |
| cache_examples=True, | |
| label='Click on any Examples below to get depth estimation results quickly 👇' | |
| ) | |
| input_img.change(infer, [input_img], [output_1, output_2, output_3]) | |
| with gr.Blocks(css=css) as demo: | |
| gr.Markdown("zoedepth") | |
| gr.Markdown("bla bla description") | |
| with gr.Tab("Depth Prediction"): | |
| create_depth_pred_demo(model) | |
| with gr.Tab("Image to 3D"): | |
| create_im_to_3d_demo(model) | |
| gr.HTML('''<br><br><br><center>You can duplicate this Space to skip the queue:<a href="https://huggingface.co/spaces/shariqfarooq/ZoeDepth?duplicate=true"><img src="https://bit.ly/3gLdBN6" alt="Duplicate Space"></a><br> | |
| <p><img src="https://visitor-badge.glitch.me/badge?page_id=shariqfarooq.zoedepth_demo_hf" alt="visitors"></p></center>''') | |
| demo.launch(debug=True, share=True) |