From b835580a78b29751c0f7f7cb9124b6061ca11988 Mon Sep 17 00:00:00 2001 From: minifull <53207038+mini-full@users.noreply.github.com> Date: Tue, 27 Feb 2024 10:10:28 +0800 Subject: [PATCH] code: allow inserting '/' as needed in img_roots Previous code did not add '/' as needed, resulting in a possible `FileNotFoundError` --- src/optimize.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/optimize.py b/src/optimize.py index e46e079..bec8f52 100644 --- a/src/optimize.py +++ b/src/optimize.py @@ -880,7 +880,7 @@ def recon(args, optimized_params, voxel_size=0.1): if scene_name == 'annotations': continue scene_names.append(scene_name) - img_roots.append(osp.join(base_root + scene_name + '/rgb')) + img_roots.append(osp.join(base_root, scene_name, 'rgb')) # ScanNet elif args.dataset_name == 'scannet_test_div_5': @@ -896,7 +896,7 @@ def recon(args, optimized_params, voxel_size=0.1): base_root = osp.join(args.gt_root, args.dataset_name) for scene_name in os.listdir(base_root): scene_names.append(scene_name) - img_roots.append(osp.join(base_root + scene_name + '/rgb')) + img_roots.append(osp.join(base_root, scene_name, 'rgb')) # TUM elif args.dataset_name == 'TUM': @@ -908,7 +908,7 @@ def recon(args, optimized_params, voxel_size=0.1): if scene_name == 'annotations': continue scene_names.append(scene_name) - img_roots.append(osp.join(base_root + scene_name + '/rgb')) + img_roots.append(osp.join(base_root, scene_name, 'rgb')) # KITTI elif args.dataset_name == 'kitti':