hofee/nbv_rec_blender_render
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix bug

Browse Source
This commit is contained in:
hofee 2024-10-21 01:01:15 +08:00
parent c2849ce9bb
commit 022fffa7c2
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
data_generator.py
View File

@ -218,8 +218,6 @@ class DataGenerator:
file_path = os.path.join(depth_dir, depth_file) file_path = os.path.join(depth_dir, depth_file)
new_file_path = os.path.join(depth_dir, f"{name}.png") new_file_path = os.path.join(depth_dir, f"{name}.png")
os.rename(file_path,new_file_path) os.rename(file_path,new_file_path)
BlenderUtils.save_blend(scene_dir)
exit(0)
return True return True

1
utils/blender_util.py
View File

@ -279,6 +279,7 @@ class BlenderUtils:
scene.render.filepath = os.path.join( scene.render.filepath = os.path.join(
output_dir, mask_dir, f"{file_name}_{cam_suffix}.png" output_dir, mask_dir, f"{file_name}_{cam_suffix}.png"
) )
scene.render.image_settings.file_format = "PNG"
scene.render.image_settings.color_depth = "8" scene.render.image_settings.color_depth = "8"
scene.render.resolution_percentage = 100 scene.render.resolution_percentage = 100
scene.render.use_overwrite = False scene.render.use_overwrite = False