From 022fffa7c29e6d4583b589d10afe964a4047dc79 Mon Sep 17 00:00:00 2001
From: hofee <lexhofee@gmail.com>
Date: Mon, 21 Oct 2024 01:01:15 +0800
Subject: [PATCH] fix bug

---
 data_generator.py     | 2 --
 utils/blender_util.py | 1 +
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/data_generator.py b/data_generator.py
index 28735a9..87a4518 100644
--- a/data_generator.py
+++ b/data_generator.py
@@ -218,8 +218,6 @@ class DataGenerator:
                 file_path = os.path.join(depth_dir, depth_file)
                 new_file_path = os.path.join(depth_dir, f"{name}.png")
                 os.rename(file_path,new_file_path)
-        BlenderUtils.save_blend(scene_dir)
-        exit(0)
         
         return True
     
diff --git a/utils/blender_util.py b/utils/blender_util.py
index 080670c..a6608f6 100644
--- a/utils/blender_util.py
+++ b/utils/blender_util.py
@@ -279,6 +279,7 @@ class BlenderUtils:
             scene.render.filepath = os.path.join(
                 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.resolution_percentage = 100
             scene.render.use_overwrite = False