mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-11 10:34:29 +08:00
25 lines
743 B
Python
25 lines
743 B
Python
from pathlib import Path
|
|
|
|
import numpy as np
|
|
|
|
from nerfstudio.data.utils.colmap_parsing_utils import Image, read_images_binary, write_images_binary
|
|
|
|
|
|
def test_multibyte_image_name_roundtrip(tmp_path: Path):
|
|
image = Image(
|
|
id=7,
|
|
qvec=np.array([1, 0, 0, 0]),
|
|
tvec=np.zeros(3),
|
|
camera_id=3,
|
|
name="室内/café.png",
|
|
xys=np.array([[2.5, 1.0]]),
|
|
point3D_ids=np.array([17]),
|
|
)
|
|
path = tmp_path / "images.bin"
|
|
write_images_binary({image.id: image}, path)
|
|
restored = read_images_binary(path)[image.id]
|
|
|
|
assert restored.name == image.name
|
|
np.testing.assert_array_equal(restored.xys, image.xys)
|
|
np.testing.assert_array_equal(restored.point3D_ids, image.point3D_ids)
|