Skip to content

Commit 2863ae1

Browse files
author
Eric Cornwell
committed
Merge branch 'arvinkx-main'
Merged 3dgrt filepath bug fix into main
2 parents aac590b + 720a20d commit 2863ae1

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

source/container/src/main.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1577,7 +1577,8 @@ def process_images(input_dir, output_dir=None):
15771577
shutil.copytree("outputs/unnamed/nerfacto/train-stage-1/nerfstudio_models", dest_dir, dirs_exist_ok=True)
15781578
elif str(config['MODEL']).lower() == "3dgut" or str(config['MODEL']).lower() == "3dgrt":
15791579
dest_dir = os.path.join(config['DATASET_PATH'], "3dgrut_models")
1580-
src_dir = os.path.join(config['DATASET_PATH'], os.listdir(os.path.join(config['DATASET_PATH'], 'exports', 'train-stage-1')[0]))
1580+
base_dir = os.path.join(config['DATASET_PATH'], 'exports', 'train-stage-1')
1581+
src_dir = os.path.join(base_dir, os.listdir(base_dir)[0])
15811582
print(f"SOURCE_DIR={src_dir}")
15821583
os.makedirs(dest_dir, exist_ok=True)
15831584
shutil.copytree(src_dir, dest_dir, dirs_exist_ok=True)

0 commit comments

Comments
 (0)