Skip to content
Snippets Groups Projects
Commit 3c7d3231 authored by Thomas Müller's avatar Thomas Müller
Browse files

scenes.py: add mip-NeRF 360 scenes

parent c1f13225
No related branches found
No related tags found
No related merge requests found
...@@ -40,6 +40,15 @@ def nerf_real_360(name, frameidx): ...@@ -40,6 +40,15 @@ def nerf_real_360(name, frameidx):
"dataset" : "", "dataset" : "",
"frameidx" : frameidx "frameidx" : frameidx
} }
def mipnerf_360(name, frameidx):
return {
"data_dir" : os.path.join(NERF_DATA_FOLDER, f"mipnerf_360/{name}"),
"dataset_train" : "transforms.json",
"dataset_test" : "transforms.json",
"dataset" : "",
"frameidx" : frameidx
}
scenes_nerf = { scenes_nerf = {
"fox" : ours_real_converted("fox/", frameidx=0), "fox" : ours_real_converted("fox/", frameidx=0),
"lego" : nerf_synthetic("lego", frameidx=52), "lego" : nerf_synthetic("lego", frameidx=52),
...@@ -54,6 +63,15 @@ scenes_nerf = { ...@@ -54,6 +63,15 @@ scenes_nerf = {
# nerf real 360 # nerf real 360
"pinecone" : nerf_real_360("pinecone", frameidx=0), "pinecone" : nerf_real_360("pinecone", frameidx=0),
"vasedeck" : nerf_real_360("vasedeck", frameidx=0), "vasedeck" : nerf_real_360("vasedeck", frameidx=0),
# mipnerf 360
"bicycle" : mipnerf_360("bicycle", frameidx=0),
"bonsai" : mipnerf_360("bonsai", frameidx=0),
"counter" : mipnerf_360("counter", frameidx=0),
"garden" : mipnerf_360("garden", frameidx=0),
"kitchen" : mipnerf_360("kitchen", frameidx=0),
"room" : mipnerf_360("room", frameidx=0),
"stump" : mipnerf_360("stump", frameidx=0),
} }
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment