#!/usr/bin/env python3 # Copyright (c) 2020-2022, NVIDIA CORPORATION. All rights reserved. # # NVIDIA CORPORATION and its licensors retain all intellectual property # and proprietary rights in and to this software, related documentation # and any modifications thereto. Any use, reproduction, disclosure or # distribution of this software and related documentation without an express # license agreement from NVIDIA CORPORATION is strictly prohibited. import os from common import * def ours_real_converted(path, frameidx): return { "data_dir" : os.path.join(NERF_DATA_FOLDER, path), "dataset_train" : "transforms.json", "dataset_test" : "transforms.json", "dataset" : "", "test_every" : 5, "frameidx" : frameidx } def nerf_synthetic(name, frameidx): return { "data_dir" : os.path.join(NERF_DATA_FOLDER, f"nerf_synthetic/{name}"), "dataset_train" : "transforms_train.json", "dataset_test" : "transforms_test.json", "dataset" : "", "frameidx" : frameidx } def nerf_real_360(name, frameidx): return { "data_dir" : os.path.join(NERF_DATA_FOLDER, f"nerf_real_360/{name}"), "dataset_train" : "transforms.json", "dataset_test" : "transforms.json", "dataset" : "", "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 = { "fox" : ours_real_converted("fox/", frameidx=0), "lego" : nerf_synthetic("lego", frameidx=52), "drums" : nerf_synthetic("drums", frameidx=52), "ship" : nerf_synthetic("ship", frameidx=52), "mic" : nerf_synthetic("mic", frameidx=52), "ficus" : nerf_synthetic("ficus", frameidx=52), "chair" : nerf_synthetic("chair", frameidx=52), "hotdog" : nerf_synthetic("hotdog", frameidx=52), "materials" : nerf_synthetic("materials", frameidx=52), # nerf real 360 "pinecone" : nerf_real_360("pinecone", 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), } def ours_mesh(name, up = [0,1,0], infolder=True): return { "data_dir" : os.path.join(SDF_DATA_FOLDER, f"{name}") if infolder else SDF_DATA_FOLDER, "dataset" : f"{name}.obj", "up" : up } scenes_sdf = { "armadillo" : ours_mesh("armadillo", infolder=False), } def ours_image(name, infolder=True): data_dir = os.path.join(IMAGE_DATA_FOLDER, f"{name}") if infolder else IMAGE_DATA_FOLDER dataset = f"{name}.bin" if not os.path.exists(os.path.join(data_dir, dataset)): dataset = f"{name}.exr" if not os.path.exists(os.path.join(data_dir, dataset)): dataset = f"{name}.png" if not os.path.exists(os.path.join(data_dir, dataset)): dataset = f"{name}.jpg" return { "data_dir" : data_dir, "dataset" : dataset } scenes_image = { "albert" : ours_image("albert", False), } def ours_volume(name, ds): return { "data_dir" : os.path.join(VOLUME_DATA_FOLDER, f"{name}"), "dataset" : ds } scenes_volume = { } def setup_colored_sdf(testbed, scene, softshadow=True): if scene == "lizard": testbed.background_color = [0.882, 0.580, 0.580, 1.000] testbed.exposure = 1.000 testbed.sun_dir=[-0.325,0.590,0.738] testbed.up_dir=[0.000,1.000,0.000] testbed.view_dir=[-0.366,-0.314,-0.876] testbed.look_at=[0.587,0.420,0.479] testbed.scale=0.887 testbed.fov,testbed.aperture_size,testbed.slice_plane_z=39.600,0.000,0.655 testbed.sdf.brdf.metallic=0.000 testbed.sdf.brdf.subsurface=0.000 testbed.sdf.brdf.specular=1.000 testbed.sdf.brdf.roughness=0.300 testbed.sdf.brdf.sheen=0.000 testbed.sdf.brdf.clearcoat=0.000 testbed.sdf.brdf.clearcoat_gloss=0.000 testbed.sdf.brdf.basecolor=[0.800,0.800,0.800] elif scene == "cow": testbed.background_color = [0.580, 0.882, 0.607, 1.000] testbed.exposure = 0.5 testbed.sun_dir=[-0.604,0.491,0.386] testbed.up_dir=[0.000,1.000,0.000] testbed.view_dir=[0.997,-0.059,-0.057] testbed.look_at=[0.500,0.500,0.500] testbed.fov,testbed.aperture_size,testbed.slice_plane_z=40.700,0.000,0.557 testbed.scale=0.976 testbed.sdf.brdf.metallic=0.000 testbed.sdf.brdf.subsurface=0.000 testbed.sdf.brdf.specular=1.000 testbed.sdf.brdf.roughness=0.300 testbed.sdf.brdf.sheen=0.000 testbed.sdf.brdf.clearcoat=0.000 testbed.sdf.brdf.clearcoat_gloss=0.000 testbed.sdf.brdf.basecolor=[0.800,0.800,0.800] elif scene == "clockwork": testbed.background_color = [0.882, 0.731, 0.580, 1.000] testbed.exposure = 3.000 testbed.sun_dir=[-0.236,0.946,-0.220] testbed.up_dir=[0.000,1.000,0.000] testbed.view_dir=[-0.639,-0.720,0.272] testbed.look_at=[0.540,0.451,0.457] testbed.scale=1.074 testbed.fov,testbed.aperture_size,testbed.slice_plane_z=39.600,0.000,0.025 testbed.sdf.brdf.metallic=1.000 testbed.sdf.brdf.subsurface=0.000 testbed.sdf.brdf.specular=1.000 testbed.sdf.brdf.roughness=0.300 testbed.sdf.brdf.sheen=0.000 testbed.sdf.brdf.clearcoat=0.000 testbed.sdf.brdf.clearcoat_gloss=0.000 testbed.sdf.brdf.basecolor=[0.800,0.800,0.800] elif scene == "lucy": testbed.background_color = [0.597, 0.797, 0.697, 1.000] testbed.exposure = 1.000 testbed.sun_dir=[0.290,0.342,0.893] testbed.up_dir=[0.000,0.000,1.000] testbed.view_dir=[0.003,-0.960,-0.281] testbed.scale=1.299 testbed.fov,testbed.aperture_size,testbed.slice_plane_z=39.600,0.000,0.768 testbed.sdf.brdf.metallic=0.000 testbed.sdf.brdf.subsurface=0.000 testbed.sdf.brdf.specular=0.194 testbed.sdf.brdf.roughness=0.300 testbed.sdf.brdf.sheen=0.000 testbed.sdf.brdf.clearcoat=0.000 testbed.sdf.brdf.clearcoat_gloss=0.000 testbed.sdf.brdf.basecolor=[0.800,0.800,0.800] softshadow = True else: if scene == "bearded_man": testbed.up_dir=[0.000,-1.000,-0.000] testbed.view_dir=[-0.924,0.128,-0.361] testbed.look_at=[0.500,0.500,0.500] testbed.fov,testbed.aperture_size,testbed.slice_plane_z=39.600,0.000,0.377 testbed.background_color = [0.580, 0.713, 0.882, 1.000] testbed.exposure = 1.000 testbed.sun_dir=[0.541,-0.839,-0.042] testbed.sdf.brdf.metallic=0.000 testbed.sdf.brdf.subsurface=0.000 testbed.sdf.brdf.specular=1.000 testbed.sdf.brdf.roughness=0.300 testbed.sdf.brdf.sheen=0.000 testbed.sdf.brdf.clearcoat=0.000 testbed.sdf.brdf.clearcoat_gloss=0.000 testbed.sdf.brdf.basecolor=[0.800,0.800,0.800] testbed.autofocus_target=[0.500,0.500,0.500] testbed.autofocus = False testbed.sdf.analytic_normals = False testbed.sdf.use_triangle_octree = False col = list(testbed.background_color) testbed.sdf.brdf.ambientcolor = np.multiply(col,col)[0:3] testbed.sdf.shadow_sharpness = 16 if softshadow else 2048 testbed.scale = testbed.scale * 1.13 def default_snapshot_filename(scene): filename = "base.msgpack" if scene["dataset"]: filename = f"{os.path.splitext(scene['dataset'])[0]}_{filename}" return os.path.join(scene["data_dir"], filename) def mode_from_scene(scene): if scene in scenes_sdf: return "sdf" elif scene in scenes_nerf: return "nerf" elif scene in scenes_image: return "image" elif scene in scenes_volume: return "volume" else: return ""