From 347d157f9c4ddace8857a2c7f7c04be920bca69e Mon Sep 17 00:00:00 2001 From: chuanfushen <77523610+chuanfushen@users.noreply.github.com> Date: Mon, 23 Jun 2025 10:02:33 +0800 Subject: [PATCH] Fixing point cloud index none issue in collate_fn.py --- opengait/data/collate_fn.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/opengait/data/collate_fn.py b/opengait/data/collate_fn.py index ae3ac4b..6767b04 100644 --- a/opengait/data/collate_fn.py +++ b/opengait/data/collate_fn.py @@ -90,7 +90,7 @@ class CollateFn(object): for i in range(feature_num): for j in indices[:self.frames_all_limit] if self.frames_all_limit > -1 and len(indices) > self.frames_all_limit else indices: - point_cloud_index = self.points_in_use.get('pointcloud_index') + point_cloud_index = self.points_in_use.get('pointcloud_index') if self.points_in_use else None if self.points_in_use is not None and point_cloud_index is not None and i == point_cloud_index: points_num = self.points_in_use.get('points_num') sample_points = (random.choices(range(len(seqs[i][j])), k=points_num) @@ -121,4 +121,4 @@ class CollateFn(object): batch[-1] = np.asarray(seqL_batch) batch[0] = fras_batch - return batch \ No newline at end of file + return batch