UniqAI/gaussian-splatting
1
0
Fork 0
mirror of https://github.com/graphdeco-inria/gaussian-splatting synced 2024-11-22 08:18:17 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

allow for subfolders in images dir

Browse Source
This commit is contained in:
Jachin 2023-10-02 15:57:32 -04:00 committed by GitHub
parent 414b553ef1
commit dcbd26dcea
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
scene/dataset_readers.py
View File

@ -94,7 +94,7 @@ def readColmapCameras(cam_extrinsics, cam_intrinsics, images_folder):
else: else:
assert False, "Colmap camera model not handled: only undistorted datasets (PINHOLE or SIMPLE_PINHOLE cameras) supported!" assert False, "Colmap camera model not handled: only undistorted datasets (PINHOLE or SIMPLE_PINHOLE cameras) supported!"
image_path = os.path.join(images_folder, os.path.basename(extr.name)) image_path = os.path.join(images_folder, extr.name)
image_name = os.path.basename(image_path).split(".")[0] image_name = os.path.basename(image_path).split(".")[0]
image = Image.open(image_path) image = Image.open(image_path)
@ -257,4 +257,4 @@ def readNerfSyntheticInfo(path, white_background, eval, extension=".png"):
sceneLoadTypeCallbacks = { sceneLoadTypeCallbacks = {
"Colmap": readColmapSceneInfo, "Colmap": readColmapSceneInfo,
"Blender" : readNerfSyntheticInfo "Blender" : readNerfSyntheticInfo
} }