download dataset

nerfstudio-project · Sep 18, 2024 · fa23297 · fa23297
1 parent e737073
commit fa23297
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 7 deletions.
diff --git a/examples/benchmarks/fisheye/mcmc_zipnerf.sh b/examples/benchmarks/fisheye/mcmc_zipnerf.sh
@@ -1,5 +1,5 @@
 SCENE_DIR="data/zipnerf"
-SCENE_LIST="berlin london nyc alameda"
+SCENE_LIST="alameda berlin london nyc"
 DATA_FACTOR=4
 RENDER_TRAJ_PATH="ellipse"
 
@@ -21,5 +21,4 @@ do
         --camera_model fisheye \
         --data_dir $SCENE_DIR/$SCENE/ \
         --result_dir $RESULT_DIR/$SCENE/
-
 done
diff --git a/examples/benchmarks/fisheye/mcmc_zipnerf_undistort.sh b/examples/benchmarks/fisheye/mcmc_zipnerf_undistort.sh
@@ -1,12 +1,12 @@
-SCENE_DIR="data/zipnerf_undistort"
-SCENE_LIST="berlin london nyc alameda"
+SCENE_DIR="data/zipnerf_undistorted"
+SCENE_LIST="alameda berlin london nyc"
 DATA_FACTOR=4
 RENDER_TRAJ_PATH="ellipse"
 
-RESULT_DIR="results/benchmark_mcmc_2M_zipnerf_undistort"
+RESULT_DIR="results/benchmark_mcmc_2M_zipnerf_undistorted"
 CAP_MAX=2000000
 
-# RESULT_DIR="results/benchmark_mcmc_4M_zipnerf_undistort"
+# RESULT_DIR="results/benchmark_mcmc_4M_zipnerf_undistorted"
 # CAP_MAX=4000000
 
 for SCENE in $SCENE_LIST;
@@ -21,5 +21,4 @@ do
         --camera_model pinhole \
         --data_dir $SCENE_DIR/$SCENE/ \
         --result_dir $RESULT_DIR/$SCENE/
-
 done
diff --git a/examples/datasets/download_dataset.py b/examples/datasets/download_dataset.py
@@ -13,20 +13,26 @@
     "mipnerf360",
     "mipnerf360_extra",
     "bilarf_data",
+    "alameda",
+    "alameda_undistorted",
 ]
 
 # dataset urls
 urls = {
     "mipnerf360": "http://storage.googleapis.com/gresearch/refraw360/360_v2.zip",
     "mipnerf360_extra": "https://storage.googleapis.com/gresearch/refraw360/360_extra_scenes.zip",
     "bilarf_data": "https://huggingface.co/datasets/Yuehao/bilarf_data/resolve/main/bilarf_data.zip",
+    "alameda": "https://storage.googleapis.com/gresearch/refraw360/zipnerf/alameda.zip",
+    "alameda_undistorted": "https://storage.googleapis.com/gresearch/refraw360/zipnerf-undistorted/alameda.zip",
 }
 
 # rename maps
 dataset_rename_map = {
     "mipnerf360": "360_v2",
     "mipnerf360_extra": "360_v2",
     "bilarf_data": "bilarf",
+    "alameda": "zipnerf/alameda",
+    "alameda_undistorted": "zipnerf_undistorted/alameda",
 }