From 34984a666e8da198a2a99d3197477027882c394b Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Sat, 17 Dec 2022 12:59:56 +0530 Subject: [PATCH] add: note on using hf datasets. --- README.md | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/README.md b/README.md index bb8cad4..a15d7c4 100644 --- a/README.md +++ b/README.md @@ -30,6 +30,12 @@ This repo offers trained models and evaluation code for the [FastDepth](http://f tar -xvf nyudepthv2.tar.gz && rm -f nyudepthv2.tar.gz cd .. ``` +- You can also load the dataset with Hugging Face Datasets with just two lines of code in Python: + ```py + from datasets import load_dataset # installation is `pip install datasets` + + nyu = load_dataset("sayakpaul/nyu_depth_v2") + ``` ## Trained Models ## The following trained models can be found at [http://datasets.lids.mit.edu/fastdepth/results/](http://datasets.lids.mit.edu/fastdepth/results/).