From 0e9259757ddb16c0208ba3327be84b107fa97831 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Linus=20H=C3=A4renstam-Nielsen?= Date: Sun, 4 Feb 2024 15:16:29 +0100 Subject: [PATCH] Fix dataloading for .xyz files --- models/dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/dataset.py b/models/dataset.py index e2842e7..6b951e8 100644 --- a/models/dataset.py +++ b/models/dataset.py @@ -21,7 +21,7 @@ def process_data(data_dir, dataname): pointcloud = trimesh.load(os.path.join(data_dir, dataname) + '.ply').vertices pointcloud = np.asarray(pointcloud) elif os.path.exists(os.path.join(data_dir, dataname) + '.xyz'): - pointcloud = np.load(os.path.join(data_dir, dataname)) + '.xyz' + pointcloud = np.load(os.path.join(data_dir, dataname) + '.xyz') else: print('Only support .xyz or .ply data. Please make adjust your data.') exit()