tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 18:44:21 +08:00
Code

Fix colmap bug when not all images are matched (#1454)

Browse Source 
Fix colmap when not 100 matching
This commit is contained in:
Matthew Tancik 2023-02-19 14:48:31 -08:00 committed by GitHub
parent ceda935e64
commit e89d113ddc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
nerfstudio/process_data/colmap_utils.py
View File

@ -604,14 +604,6 @@ def colmap_to_json(
cameras = read_cameras_binary(cameras_path)
images = read_images_binary(images_path)
# Images were renamed to frame_{i:05d}.{ext} and
# the filenames needs to be replaced in the transforms.json as well
original_filenames = [x.name for x in images.values()]
# Sort was used in nerfstudio.process_data.process_data_utils:get_image_filenames
original_filenames.sort()
# Build the map to the new filenames
filename_map = {name: f"frame_{i+1:05d}{os.path.splitext(name)[-1]}" for i, name in enumerate(original_filenames)}
# Only supports one camera
camera_params = cameras[1].params
@ -627,8 +619,7 @@ def colmap_to_json(
c2w = c2w[np.array([1, 0, 2, 3]), :]
c2w[2, :] *= -1
name = filename_map[im_data.name]
name = Path(f"./images/{name}")
name = Path(f"./images/{im_data.name}")
frame = {
"file_path": name.as_posix(),