Merge pull request princeton-vl#273 from princeton-vl/export_fix

fix export bug when optimize_terrain_diskusage is on; fix input folde…

infinigen/core/execute_tasks.py

@@ -308,7 +308,7 @@ def execute_tasks(
         col.hide_viewport = False
 
     if need_terrain_processing and (
-        Task.Render in task or Task.GroundTruth in task or Task.MeshSave in task
+        Task.Render in task or Task.GroundTruth in task or Task.MeshSave in task or Task.Export in task
     ):
         terrain = Terrain(
             scene_seed,

infinigen/datagen/job_funcs.py

@@ -122,7 +122,21 @@ def queue_export(
     **kwargs,
 ):
     input_suffix = get_suffix(input_indices)
-    input_folder = f"{folder}/coarse{input_suffix}"
+    input_folder_priority_options = [
+        f"fine{input_suffix}",
+        "fine",
+        f"coarse{input_suffix}",
+        "coarse",
+    ]
+
+    for option in input_folder_priority_options:
+        input_folder = f"{folder}/{option}"
+        if (Path(input_folder) / "scene.blend").exists():
+            break
+    else:
+        logger.warning(
+            f"No scene.blend found in {input_folder} for any of {input_folder_priority_options}"
+        )
 
     cmd = (
         get_cmd(

infinigen/tools/export.py

@@ -761,7 +761,6 @@ def export_scene(
     task_uniqname=None,
     **kwargs,
 ):
-    bpy.ops.wm.open_mainfile(filepath=str(input_blend))
     folder = output_folder / f"export_{input_blend.name}"
     folder.mkdir(exist_ok=True, parents=True)
     export_curr_scene(folder, **kwargs)