Fixed load_config path bug in container

Eric Cornwell · Eric Cornwell · commit aac590b7371a · 2025-07-23T02:44:54.000-05:00
diff --git a/source/container/src/main.py b/source/container/src/main.py
@@ -1158,12 +1158,11 @@ def process_images(input_dir, output_dir=None):
                     requires_gpu=False
                 )
             else:
-                config_path = f"outputs/unnamed/{str(config['MODEL']).lower()}/train-stage-1/config.yml"
                 if str(config['MODEL']).lower() == "nerfacto":
                     # Geometry
                     args = [
                         "poisson",
-                        "--load-config", config_path, #"outputs/unnamed/nerfacto/train-stage-1/config.yml",
+                        "--load-config", "outputs/unnamed/nerfacto/train-stage-1/config.yml",
                         "--output-dir", output_path
                     ]
                     pipeline.create_component(
@@ -1177,7 +1176,7 @@ def process_images(input_dir, output_dir=None):
                     )
                     # Texture
                     args = [
-                        "--load-config", config_path, #"outputs/unnamed/nerfacto/train-stage-1/config.yml",
+                        "--load-config", "outputs/unnamed/nerfacto/train-stage-1/config.yml",
                         "--input-mesh-filename", os.path.join(output_path, "poisson_mesh.ply"),
                         "--output-dir", os.path.join(output_path, "textured")
                     ]
@@ -1192,7 +1191,7 @@ def process_images(input_dir, output_dir=None):
                     )
                 elif str(config['MODEL']).lower() == "splatfacto-w-light":
                     args = [
-                        "--load_config", config_path, #"outputs/unnamed/splatfacto-w-light/train-stage-1/config.yml",
+                        "--load_config", "outputs/unnamed/splatfacto-w-light/train-stage-1/config.yml",
                         "--output_dir", output_path,
                         "--camera_idx", "0" #str(math.ceil(float(config['MAX_NUM_IMAGES'])/2))
                     ]
@@ -1208,7 +1207,7 @@ def process_images(input_dir, output_dir=None):
                 else:
                     args = [
                         "gaussian-splat",
-                        "--load-config", config_path, #"outputs/unnamed/splatfacto/train-stage-1/config.yml",
+                        "--load-config", "outputs/unnamed/splatfacto/train-stage-1/config.yml",
                         "--output-dir", output_path
                     ]
                     pipeline.create_component(

Original file line number	Diff line number	Diff line change
`@@ -1158,12 +1158,11 @@ def process_images(input_dir, output_dir=None):`
`1158`	`1158`	`requires_gpu=False`
`1159`	`1159`	`)`
`1160`	`1160`	`else:`
`1161`		`- config_path = f"outputs/unnamed/{str(config['MODEL']).lower()}/train-stage-1/config.yml"`
`1162`	`1161`	`if str(config['MODEL']).lower() == "nerfacto":`
`1163`	`1162`	`# Geometry`
`1164`	`1163`	`args = [`
`1165`	`1164`	`"poisson",`
`1166`		`- "--load-config", config_path, #"outputs/unnamed/nerfacto/train-stage-1/config.yml",`
	`1165`	`+ "--load-config", "outputs/unnamed/nerfacto/train-stage-1/config.yml",`
`1167`	`1166`	`"--output-dir", output_path`
`1168`	`1167`	`]`
`1169`	`1168`	`pipeline.create_component(`
`@@ -1177,7 +1176,7 @@ def process_images(input_dir, output_dir=None):`
`1177`	`1176`	`)`
`1178`	`1177`	`# Texture`
`1179`	`1178`	`args = [`
`1180`		`- "--load-config", config_path, #"outputs/unnamed/nerfacto/train-stage-1/config.yml",`
	`1179`	`+ "--load-config", "outputs/unnamed/nerfacto/train-stage-1/config.yml",`
`1181`	`1180`	`"--input-mesh-filename", os.path.join(output_path, "poisson_mesh.ply"),`
`1182`	`1181`	`"--output-dir", os.path.join(output_path, "textured")`
`1183`	`1182`	`]`
`@@ -1192,7 +1191,7 @@ def process_images(input_dir, output_dir=None):`
`1192`	`1191`	`)`
`1193`	`1192`	`elif str(config['MODEL']).lower() == "splatfacto-w-light":`
`1194`	`1193`	`args = [`
`1195`		`- "--load_config", config_path, #"outputs/unnamed/splatfacto-w-light/train-stage-1/config.yml",`
	`1194`	`+ "--load_config", "outputs/unnamed/splatfacto-w-light/train-stage-1/config.yml",`
`1196`	`1195`	`"--output_dir", output_path,`
`1197`	`1196`	`"--camera_idx", "0" #str(math.ceil(float(config['MAX_NUM_IMAGES'])/2))`
`1198`	`1197`	`]`
`@@ -1208,7 +1207,7 @@ def process_images(input_dir, output_dir=None):`
`1208`	`1207`	`else:`
`1209`	`1208`	`args = [`
`1210`	`1209`	`"gaussian-splat",`
`1211`		`- "--load-config", config_path, #"outputs/unnamed/splatfacto/train-stage-1/config.yml",`
	`1210`	`+ "--load-config", "outputs/unnamed/splatfacto/train-stage-1/config.yml",`
`1212`	`1211`	`"--output-dir", output_path`
`1213`	`1212`	`]`
`1214`	`1213`	`pipeline.create_component(`