43 lines
1.6 KiB
Python
43 lines
1.6 KiB
Python
|
|
||
|
|
import os
|
||
|
|
import shutil
|
||
|
|
import json
|
||
|
|
import subprocess
|
||
|
|
import tempfile
|
||
|
|
|
||
|
|
|
||
|
|
class ViewRenderUtil:
|
||
|
|
blender_path = r"C:\Program Files\Blender Foundation\Blender 4.0\blender.exe"
|
||
|
|
@staticmethod
|
||
|
|
def render_view(cam_pose, scene_path, script_path):
|
||
|
|
|
||
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
||
|
|
params = {
|
||
|
|
"cam_pose": cam_pose.tolist(),
|
||
|
|
"scene_path": scene_path
|
||
|
|
}
|
||
|
|
scene_info_path = os.path.join(scene_path, "scene_info.json")
|
||
|
|
shutil.copy(scene_info_path, os.path.join(temp_dir, "scene_info.json"))
|
||
|
|
params_data_path = os.path.join(temp_dir, "params.json")
|
||
|
|
with open(params_data_path, 'w') as f:
|
||
|
|
json.dump(params, f)
|
||
|
|
import ipdb; ipdb.set_trace()
|
||
|
|
result = subprocess.run([
|
||
|
|
ViewRenderUtil.blender_path, '-b', '-P', script_path, '--', temp_dir
|
||
|
|
], capture_output=True, text=True)
|
||
|
|
print(result.stdout)
|
||
|
|
print(result.stderr)
|
||
|
|
path = os.path.join(temp_dir, "tmp")
|
||
|
|
|
||
|
|
return None
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
import numpy as np
|
||
|
|
idx = 0
|
||
|
|
cam_param_path = r"D:\Project\nbv_rec\data\google_scan-backpack_0288\camera_params\{}.json"
|
||
|
|
cam_pose = json.load(open(cam_param_path.format(idx)))
|
||
|
|
cam_pose = np.array(cam_pose["extrinsic"])
|
||
|
|
scene_path = r"D:\Project\nbv_rec\data\google_scan-backpack_0288"
|
||
|
|
script_path = r"D:\Project\nbv_rec\nbv_rec_blender_render\data_renderer.py"
|
||
|
|
ViewRenderUtil.render_view(cam_pose, scene_path, script_path)
|