allow for subfolders in images dir

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

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)