mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-15 04:13:57 +08:00
Deadline results
This commit is contained in:
frfoc1 committed
1 parent
f3d34d442f
commit
8ee91791d9
5 files changed
+325
-2
No files matched your search
@@ -0,0 +1,58 @@
|
||||
"""
|
||||
Script to render IBL renders
|
||||
Usage: blender <blend file> -P hdr_blender.py -b
|
||||
"""
|
||||
import os
|
||||
import random
|
||||
from math import radians
|
||||
from os import listdir
|
||||
from os.path import isfile, join
|
||||
|
||||
import bpy
|
||||
import cycles
|
||||
from mathutils import Euler, Matrix, Vector
|
||||
|
||||
|
||||
import sys
|
||||
argv = sys.argv
|
||||
argv = argv[argv.index("--") + 1:] # get all args after "--"
|
||||
|
||||
inp_dirs = [argv[0]]
|
||||
out_dirs = [argv[1]]
|
||||
|
||||
pairs = list(zip(inp_dirs,out_dirs))
|
||||
scn = bpy.context.scene
|
||||
scn.render.engine = 'CYCLES'
|
||||
scn.render.image_settings.file_format = 'OPEN_EXR'
|
||||
scn.world.use_nodes = True
|
||||
# Enabf_le GPU rendering.
|
||||
# bpy.context.preferences.addons['cycles'].preferences.compute_device_type = 'CUDA'
|
||||
bpy.context.preferences.addons['cycles'].preferences.get_devices()
|
||||
for device in bpy.context.preferences.addons['cycles'].preferences.devices:
|
||||
if "TITAN RX" == device.name:
|
||||
device.use = True
|
||||
else:
|
||||
device.use = False
|
||||
# select world node tree
|
||||
wd = scn.world
|
||||
nt = bpy.data.worlds[wd.name].node_tree
|
||||
|
||||
for pair in pairs:
|
||||
inp, out = pair
|
||||
inp = inp
|
||||
os.makedirs(out,exist_ok=True)
|
||||
S = bpy.context.scene
|
||||
envFiles = [f for f in listdir(inp) if isfile(join(inp, f)) and f.endswith('.exr')]
|
||||
i = 0
|
||||
for env_map in envFiles:
|
||||
env_file = join(inp, env_map)
|
||||
bpy.ops.image.open(filepath=env_file, directory=inp, files=[
|
||||
{"name": env_map, "name": env_map}], show_multiview=False)
|
||||
output_name = env_map.split('.')[0] +'.exr'
|
||||
|
||||
backNode = nt.nodes['Environment Texture']
|
||||
backNode.image = bpy.data.images.load(env_file)
|
||||
|
||||
bpy.context.scene.render.filepath = os.path.join(out, output_name)
|
||||
print("Output",os.path.join(out, output_name))
|
||||
bpy.ops.render.render(write_still=True)
|
||||
@@ -0,0 +1,160 @@
|
||||
|
||||
export scene_name=my_apartment_ns
|
||||
export output_dir=renders/${scene_name}/lantern-nerfacto/video_images
|
||||
export root_dir=data/${scene_name}
|
||||
export model_path=outputs/${scene_name}/lantern-nerfacto/2024-05-16_212421/config.yml
|
||||
|
||||
# camera paths
|
||||
# video ground: data/Henry_Gagnon_blue_et_rouge_ns/camera_paths/2024-05-16_015020.json
|
||||
# still bench: 2024-05-16_015020_render
|
||||
# pespective cam: 2024-05-16_015020_render3
|
||||
|
||||
# meeting room
|
||||
# export camera_path_name=2024-05-16_201826_persp
|
||||
# export camera_path_name=2024-05-16_201826_envmap
|
||||
|
||||
# my_apartment
|
||||
# export camera_path_name=2024-05-16_212421_persp
|
||||
# export camera_path_name=2024-05-16_212421_env1
|
||||
# export camera_path_name=2024-05-16_212421_env2
|
||||
# export camera_path_name=2024-05-16_212421_env3
|
||||
# export camera_path_name=2024-05-16_212421_env4
|
||||
export camera_path_name=2024-05-16_212421_env5
|
||||
export camera_path=data/${scene_name}/camera_paths/${camera_path_name}.json
|
||||
|
||||
ns-render camera-path --load-config $model_path --camera-path-filename $camera_path --output-path $output_dir/${camera_path_name}/well-expo/ --output-format images
|
||||
ns-render camera-path --load-config $model_path --camera-path-filename $camera_path --output-path $output_dir/${camera_path_name}/fast-expo/ --output-format images --rendered_output_names rgb_fast
|
||||
|
||||
echo "Combine two exposures"
|
||||
python video_rendering_code/combine.py --well_dir $output_dir/${camera_path_name}/well-expo/ --fast_dir $output_dir/${camera_path_name}/fast-expo/ --out_dir $output_dir/${camera_path_name}/combined/ --do_linearization --experiment_location $root_dir
|
||||
|
||||
# blender --background video_rendering_code/scene_new.blend -P video_rendering_code/hdr_blender.py -- $output_dir/${camera_path_name}/combined/ $output_dir/${camera_path_name}/render/
|
||||
|
||||
|
||||
|
||||
|
||||
#########
|
||||
|
||||
export camera_path_name=2024-05-16_212421_env1
|
||||
export camera_path=data/${scene_name}/camera_paths/${camera_path_name}.json
|
||||
|
||||
echo "Combine two exposures"
|
||||
python video_rendering_code/combine.py --well_dir $output_dir/${camera_path_name}/well-expo/ --fast_dir $output_dir/${camera_path_name}/fast-expo/ --out_dir $output_dir/${camera_path_name}/combined/ --do_linearization --experiment_location $root_dir
|
||||
|
||||
|
||||
export camera_path_name=2024-05-16_212421_env2
|
||||
export camera_path=data/${scene_name}/camera_paths/${camera_path_name}.json
|
||||
|
||||
echo "Combine two exposures"
|
||||
python video_rendering_code/combine.py --well_dir $output_dir/${camera_path_name}/well-expo/ --fast_dir $output_dir/${camera_path_name}/fast-expo/ --out_dir $output_dir/${camera_path_name}/combined/ --do_linearization --experiment_location $root_dir
|
||||
|
||||
|
||||
|
||||
|
||||
export camera_path_name=2024-05-16_212421_env3
|
||||
export camera_path=data/${scene_name}/camera_paths/${camera_path_name}.json
|
||||
|
||||
echo "Combine two exposures"
|
||||
python video_rendering_code/combine.py --well_dir $output_dir/${camera_path_name}/well-expo/ --fast_dir $output_dir/${camera_path_name}/fast-expo/ --out_dir $output_dir/${camera_path_name}/combined/ --do_linearization --experiment_location $root_dir
|
||||
|
||||
|
||||
|
||||
|
||||
export camera_path_name=2024-05-16_212421_env4
|
||||
export camera_path=data/${scene_name}/camera_paths/${camera_path_name}.json
|
||||
|
||||
echo "Combine two exposures"
|
||||
python video_rendering_code/combine.py --well_dir $output_dir/${camera_path_name}/well-expo/ --fast_dir $output_dir/${camera_path_name}/fast-expo/ --out_dir $output_dir/${camera_path_name}/combined/ --do_linearization --experiment_location $root_dir
|
||||
export camera_path_name=2024-05-16_212421_env5
|
||||
export camera_path=data/${scene_name}/camera_paths/${camera_path_name}.json
|
||||
|
||||
echo "Combine two exposures"
|
||||
python video_rendering_code/combine.py --well_dir $output_dir/${camera_path_name}/well-expo/ --fast_dir $output_dir/${camera_path_name}/fast-expo/ --out_dir $output_dir/${camera_path_name}/combined/ --do_linearization --experiment_location $root_dir
|
||||
|
||||
|
||||
####
|
||||
|
||||
# meeting room teaser
|
||||
|
||||
|
||||
export scene_name=meeting_room_ns
|
||||
export output_dir=renders/${scene_name}/lantern-nerfacto/video_images
|
||||
export root_dir=data/${scene_name}
|
||||
export model_path=outputs/${scene_name}/lantern-nerfacto/2024-05-16_201826/config.yml
|
||||
|
||||
# meeting room
|
||||
# export camera_path_name=2024-05-16_201826_persp
|
||||
# export camera_path_name=2024-05-16_201826_persp2
|
||||
|
||||
#export camera_path_name=2024-05-16_201826_env1
|
||||
# export camera_path_name=2024-05-16_201826_env2
|
||||
# export camera_path_name=2024-05-16_201826_env3
|
||||
# export camera_path_name=2024-05-16_201826_env4
|
||||
# export camera_path_name=2024-05-16_201826_env5
|
||||
# export camera_path_name=2024-05-16_201826_env6
|
||||
# export camera_path_name=2024-05-16_201826_env7
|
||||
export camera_path_name=2024-05-16_201826_env8
|
||||
|
||||
export camera_path=data/${scene_name}/camera_paths/${camera_path_name}.json
|
||||
|
||||
ns-render camera-path --load-config $model_path --camera-path-filename $camera_path --output-path $output_dir/${camera_path_name}/well-expo/ --output-format images
|
||||
ns-render camera-path --load-config $model_path --camera-path-filename $camera_path --output-path $output_dir/${camera_path_name}/fast-expo/ --output-format images --rendered_output_names rgb_fast
|
||||
|
||||
echo "Combine two exposures"
|
||||
python lantern_scripts/combine.py --well_dir $output_dir/${camera_path_name}/well-expo/ --fast_dir $output_dir/${camera_path_name}/fast-expo/ --out_dir $output_dir/${camera_path_name}/combined/ --do_linearization --experiment_location $root_dir
|
||||
|
||||
# Fig. 5 - ABO objects
|
||||
|
||||
export scene_name=LVSN_LAB_OpenSFM
|
||||
export output_dir=renders/${scene_name}/lantern-nerfacto/video_images
|
||||
export root_dir=data/${scene_name}
|
||||
export model_path=outputs/${scene_name}/lantern-nerfacto/step_2/config.yml
|
||||
|
||||
# export camera_path_name=persp1
|
||||
export camera_path_name=persp2
|
||||
# export camera_path_name=env1
|
||||
# export camera_path_name=env2
|
||||
# export camera_path_name=env3
|
||||
|
||||
export camera_path=data/${scene_name}/camera_paths/${camera_path_name}.json
|
||||
|
||||
ns-render camera-path --load-config $model_path --camera-path-filename $camera_path --output-path $output_dir/${camera_path_name}/well-expo/ --output-format images
|
||||
ns-render camera-path --load-config $model_path --camera-path-filename $camera_path --output-path $output_dir/${camera_path_name}/fast-expo/ --output-format images --rendered_output_names rgb_fast
|
||||
|
||||
echo "Combine two exposures"
|
||||
python lantern_scripts/combine.py --well_dir $output_dir/${camera_path_name}/well-expo/ --fast_dir $output_dir/${camera_path_name}/fast-expo/ --out_dir $output_dir/${camera_path_name}/combined/ --do_linearization --experiment_location $root_dir
|
||||
|
||||
######### Render videos ############
|
||||
|
||||
# LVSN lab upstairs
|
||||
|
||||
|
||||
export scene_name=LVSN_LAB_OpenSFM
|
||||
export model_type=hdr-nerfacto
|
||||
# export model_path=outputs/${scene_name}/lantern-nerfacto/step_2/config.yml
|
||||
export output_dir=renders/${scene_name}/${model_type}/video_images
|
||||
export model_path=outputs/${scene_name}/${model_type}/2024-05-18_143132/config.yml
|
||||
export camera_path_name=video3
|
||||
export camera_path=data/${scene_name}/camera_paths/${camera_path_name}.json
|
||||
|
||||
ns-render camera-path --load-config $model_path --camera-path-filename $camera_path --output-path $output_dir/${camera_path_name}/well-expo/ --output-format images
|
||||
ns-render camera-path --load-config $model_path --camera-path-filename $camera_path --output-path $output_dir/${camera_path_name}/fast-expo/ --output-format images --rendered_output_names rgb_fast
|
||||
|
||||
echo "Combine two exposures"
|
||||
python lantern_scripts/combine.py --well_dir $output_dir/${camera_path_name}/well-expo/ --fast_dir $output_dir/${camera_path_name}/fast-expo/ --out_dir $output_dir/${camera_path_name}/combined/ --do_linearization --experiment_location $root_dir
|
||||
|
||||
# for Yannick
|
||||
zip -rq combined_${scene_name}_${model_type}.zip $output_dir/${camera_path_name}/combined
|
||||
|
||||
blender --background video_rendering_code/scene_new.blend -P video_rendering_code/hdr_blender.py -- $output_dir/${camera_path_name}/combined/ $output_dir/${camera_path_name}/render/
|
||||
|
||||
mkdir $output_dir/${camera_path_name}/render_tonemapped
|
||||
python lantern_scripts/tonemap.py --data_dir $output_dir/${camera_path_name}/render/ --out_dir $output_dir/${camera_path_name}/render_tonemapped
|
||||
python lantern_scripts/mp4_maker.py --input_Addr $output_dir/${camera_path_name}/render_tonemapped --out_file $output_dir/${camera_path_name}/video_render.mp4
|
||||
|
||||
##
|
||||
|
||||
export scene_name=meeting_room_ns
|
||||
|
||||
export model_path=outputs/${scene_name}/lantern-nerfacto/2024-05-16_201826/config.yml
|
||||
|
||||
Reference in new issue
Block a user