diff --git a/datasets/ngp.py b/datasets/ngp.py index 36f7861..ab46574 100644 --- a/datasets/ngp.py +++ b/datasets/ngp.py @@ -52,13 +52,14 @@ def read_meta(self, split): for frame in tqdm(frames): c2w = np.array(frame['transform_matrix'])[:3, :4] - self.poses += [c2w] + try: img_path = os.path.join(self.root_dir, f"{frame['file_path']}") img = read_image(img_path, self.img_wh) self.rays += [img] + self.poses += [c2w] except: print("can not read image")