File size: 588 Bytes
783f492 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
import gradio as gr
from datasets import load_dataset
dataset = load_dataset("sayakpaul/nyu_depth_v2", split="train[:2%]", trust_remote_code=True)
def preview(index):
sample = dataset[int(index)]
return sample["image"], sample["depth"], sample["label"]
gr.Interface(
fn=preview,
inputs=gr.Slider(0, len(dataset)-1, step=1, label="Sample Index"),
outputs=[
gr.Image(type="pil", label="RGB Image"),
gr.Image(type="numpy", label="Depth Map"),
gr.Image(type="numpy", label="Segmentation Label")
],
title="NYU Depth V2 Viewer"
).launch()
|