mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-12 02:45:34 +08:00
Sasha Mitchell
2f91cb7ad9
read_images_text kept only the first word, so "my photo.jpg" came back as "my".
30 lines
880 B
Python
30 lines
880 B
Python
"""COLMAP images.txt names can contain spaces."""
|
|
|
|
from pathlib import Path
|
|
|
|
from nerfstudio.data.utils.colmap_parsing_utils import read_images_text
|
|
|
|
|
|
def test_image_name_keeps_spaces(tmp_path: Path):
|
|
images_txt = tmp_path / "images.txt"
|
|
images_txt.write_text(
|
|
"\n".join(
|
|
[
|
|
"# Image list with two lines of data per image:",
|
|
"1 1.0 0.0 0.0 0.0 0.0 0.0 0.0 1 my photo.jpg",
|
|
"1.5 2.5 7",
|
|
"2 1.0 0.0 0.0 0.0 0.0 0.0 0.0 1 plain.jpg",
|
|
"0.0 0.0 -1",
|
|
"",
|
|
]
|
|
)
|
|
)
|
|
|
|
images = read_images_text(images_txt)
|
|
|
|
assert images[1].name == "my photo.jpg"
|
|
assert images[1].point3D_ids.tolist() == [7]
|
|
assert images[1].xys.tolist() == [[1.5, 2.5]]
|
|
assert images[2].name == "plain.jpg"
|
|
assert images[2].point3D_ids.tolist() == [-1]
|