fix bug

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
     

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